prevent loading of properties when training without conditions

atomistic-machine-learning · Jun 1, 2023 · ba0b526 · ba0b526
1 parent acbe70e
commit ba0b526
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/schnetpack_gschnet/data/datamodule.py b/src/schnetpack_gschnet/data/datamodule.py
@@ -407,7 +407,7 @@ def load_partitions(self):
                 )
 
     def register_properties(self, properties: List[str]):
-        if properties is not None and len(properties) > 0:
+        if properties is not None:
             available_properties = self.dataset.available_properties
             for p in properties:
                 if p not in available_properties:

diff --git a/src/schnetpack_gschnet/model.py b/src/schnetpack_gschnet/model.py
@@ -304,7 +304,7 @@ def predict_distances(
 
     def get_required_data_properties(self):
         if self.conditioning_module is None:
-            return None
+            return []
         else:
             return self.conditioning_module.required_data_properties