Fix floor length check

MLSTRUCT · Dec 8, 2024 · 1be1cb1 · 1be1cb1
1 parent 5cdcadb
commit 1be1cb1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/MLStructFP_benchmarks/utils/_fp_db.py b/MLStructFP_benchmarks/utils/_fp_db.py
@@ -132,7 +132,11 @@ def process_dataset(
                     'respectively')
         assert '/' in path, path_err
 
-        t = len(db.floors)
+        t: int = len(db.floors)  # Total floors
+        if t == 0:
+            print('There are no floors. Process finished')
+            return []
+
         print(f'Total floors to compute in parallel: {t}')
         print(f'Using up to {num_proc}/{cpu_count()} CPUs')
         print(f'Using export path: {path}, image size: {isz}px, patch size: {psz}m')