fix

ACEsuit · Feb 3, 2023 · bbfc299 · bbfc299
1 parent f5bbc34
commit bbfc299
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mace_jax/tools/gin_datasets.py b/mace_jax/tools/gin_datasets.py
@@ -41,7 +41,7 @@ def datasets(
     """Load training and test dataset from xyz file"""
 
     atomic_energies_dict, all_train_configs = data.load_from_xyz(
-        file_path=train_path,
+        file_or_path=train_path,
         config_type_weights=config_type_weights,
         energy_key=energy_key,
         forces_key=forces_key,
@@ -56,7 +56,7 @@ def datasets(
 
     if valid_path is not None:
         _, valid_configs = data.load_from_xyz(
-            file_path=valid_path,
+            file_or_path=valid_path,
             config_type_weights=config_type_weights,
             energy_key=energy_key,
             forces_key=forces_key,
@@ -88,7 +88,7 @@ def datasets(
 
     if test_path is not None:
         _, test_configs = data.load_from_xyz(
-            file_path=test_path,
+            file_or_path=test_path,
             config_type_weights=config_type_weights,
             energy_key=energy_key,
             forces_key=forces_key,