bugs fixed, binary datasets working

2021-03-27 18:23:51 +01:00
parent 37e36df0a8
commit 76eb567eed
4 changed files with 33 additions and 44 deletions
--- a/datasets/compare_base.py
+++ b/datasets/compare_base.py
@@ -144,7 +144,9 @@ class CompareBase(_BaseDataModule):
                    print(f'{data_option} skipped...')
                    continue

-            if lab_file is not None:
+            if lab_file is None:
+                lab_file = f'{data_option}.csv'
+            elif lab_file is not None:
                if any([x in lab_file for x in data_options]):
                    lab_file = f'{data_option}.csv'