Merge pull request #74 from kapsner/main

fix: downgrading torchmetrics due to import error

nndet/utils/config.py

@@ -39,11 +39,15 @@ def load_dataset_info(task_dir: Pathlike) -> dict:
     """
     task_dir = Path(task_dir)
     yaml_path = task_dir / "dataset.yaml"
+    yaml_path_fallback = task_dir / "dataset.yml"
     json_path = task_dir / "dataset.json"
 
     if yaml_path.is_file():
         with open(yaml_path, 'r') as f:
             data = yaml.full_load(f)
+    elif yaml_path_fallback.is_file():
+        with open(yaml_path_fallback, 'r') as f:
+            data = yaml.full_load(f)
     elif json_path.is_file():
         with open(json_path, "r") as f:
             data = json.load(f)

requirements.txt

@@ -16,4 +16,5 @@ hydra-core>=1.1.0
 mlflow
 GitPython
 matplotlib
-seaborn
\ No newline at end of file
+seaborn
+torchmetrics>=0.7.0,<=0.7.3