Merge pull request #219 from melo-gonzo/checkpoint-check-bugfix

Checkpoint Path Check Bugfix

matsciml/interfaces/ase/base.py

@@ -68,7 +68,7 @@ def recursive_type_cast(
     return data_dict
 
 
-def __checkpoint_conversion_exist_check(ckpt_path: str | Path) -> Path:
+def _checkpoint_conversion_exist_check(ckpt_path: str | Path) -> Path:
     """Standardizes and checks for checkpoint path existence."""
     if isinstance(ckpt_path, str):
         ckpt_path = Path(ckpt_path)
@@ -261,22 +261,22 @@ def calculate(
     def from_pretrained_force_regression(
         cls, ckpt_path: str | Path, *args, **kwargs
     ) -> MatSciMLCalculator:
-        ckpt_path = __checkpoint_conversion_exist_check(ckpt_path)
+        ckpt_path = _checkpoint_conversion_exist_check(ckpt_path)
         task = ForceRegressionTask.load_from_checkpoint(ckpt_path)
         return cls(task, *args, **kwargs)
 
     @classmethod
     def from_pretrained_gradfree_task(
         cls, ckpt_path: str | Path, *args, **kwargs
     ) -> MatSciMLCalculator:
-        ckpt_path = __checkpoint_conversion_exist_check(ckpt_path)
+        ckpt_path = _checkpoint_conversion_exist_check(ckpt_path)
         task = GradFreeForceRegressionTask.load_from_checkpoint(ckpt_path)
         return cls(task, *args, **kwargs)
 
     @classmethod
     def from_pretrained_scalar_task(
         cls, ckpt_path: str | Path, *args, **kwargs
     ) -> MatSciMLCalculator:
-        ckpt_path = __checkpoint_conversion_exist_check(ckpt_path)
+        ckpt_path = _checkpoint_conversion_exist_check(ckpt_path)
         task = ScalarRegressionTask.load_from_checkpoint(ckpt_path)
         return cls(task, *args, **kwargs)