Allow compute_distributions() to work with filepaths already with a .…

…json extension (#846)
v7labs · May 22, 2024 · 1f42966 · 1f42966
1 parent 19bc2a4
commit 1f42966
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/darwin/dataset/utils.py b/darwin/dataset/utils.py
@@ -797,7 +797,9 @@ def compute_distributions(
             stems: List[str] = [e.rstrip("\n\r") for e in split_file.open()]
 
             for stem in stems:
-                annotation_path: Path = annotations_dir / f"{stem}.json"
+                if not stem.endswith(".json"):
+                    stem = f"{stem}.json"
+                annotation_path: Path = annotations_dir / stem
                 annotation_file: Optional[dt.AnnotationFile] = parse_path(
                     annotation_path
                 )