Hotfix/compute distributions (#339)

* fallback to random split * add setup.py Co-authored-by: Andrea Azzini <[email protected]>
v7labs · Feb 8, 2022 · 967991b · 967991b
1 parent 9cc6588
commit 967991b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/darwin/dataset/utils.py b/darwin/dataset/utils.py
@@ -496,6 +496,8 @@ def compute_distributions(
     for partition in partitions:
         for annotation_type in annotation_types:
             split_file: Path = split_path / f"stratified_{annotation_type}_{partition}.txt"
+            if not split_file.exists():
+                split_file = split_path / f"random_{partition}.txt"
             stems: List[str] = [e.rstrip("\n\r") for e in split_file.open()]
 
             for stem in stems:

diff --git a/setup.py b/setup.py
@@ -47,6 +47,7 @@
         "darwin.exporter",
         "darwin.importer.formats",
         "darwin.exporter.formats",
+        "darwin.version",
     ],
     entry_points={"console_scripts": ["darwin=darwin.cli:main"]},
     classifiers=["Programming Language :: Python :: 3", "License :: OSI Approved :: MIT License"],