Fix type hint in dataset.py (#55)

grok-ai · Feb 15, 2022 · 41ff1e9 · 41ff1e9
1 parent 052aae5
commit 41ff1e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/data/dataset.py b/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/data/dataset.py
@@ -43,7 +43,7 @@ def main(cfg: omegaconf.DictConfig) -> None:
     Args:
         cfg: the hydra configuration
     """
-    _: MyDataset = hydra.utils.instantiate(cfg.nn.data.datasets.train, split="train", _recursive_=False)
+    _: Dataset = hydra.utils.instantiate(cfg.nn.data.datasets.train, split="train", _recursive_=False)
 
 
 if __name__ == "__main__":