Ensure that FileParameter is used for save_name

fjclark · May 11, 2024 · d0671dc · d0671dc
1 parent 134720f
commit d0671dc
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/maize/graphs/exs/biosimspace/system_preparation.py b/maize/graphs/exs/biosimspace/system_preparation.py
@@ -83,7 +83,7 @@ class _SystemPreparationBase(Graph, ABC):
     runtime_unrestrained_npt: Parameter[float] = Parameter(default=1.0)
     """The runtime for NPT equilibration without restraints, in ns"""
 
-    save_name: Parameter[Path] = Parameter(default="bss_system")
+    save_name: FileParameter[Path] = FileParameter(default="bss_system")
 
     def build(self) -> None:
         param_lig = self.add(Parameterise, name="ParameteriseLigand")

diff --git a/maize/steps/exs/biosimspace/equilibrate.py b/maize/steps/exs/biosimspace/equilibrate.py
@@ -8,7 +8,7 @@
 
 import pytest
 
-from maize.core.interface import Parameter
+from maize.core.interface import FileParameter, Parameter
 from maize.utilities.testing import TestRig
 
 from ._base import _BioSimSpaceBase
@@ -98,7 +98,7 @@ class _EquilibrateBase(_BioSimSpaceBase, ABC):
     angstrom**2.
     """
 
-    save_name: Parameter[Path] = Parameter(optional=True)
+    save_name: FileParameter[Path] = FileParameter(optional=True)
     """
     If supplied, the output files will be saved with
     this name. E.g., if set to Path("output"), the output

diff --git a/maize/steps/exs/biosimspace/minimise.py b/maize/steps/exs/biosimspace/minimise.py
@@ -8,7 +8,7 @@
 
 import pytest
 
-from maize.core.interface import Parameter
+from maize.core.interface import FileParameter, Parameter
 from maize.core.workflow import expose
 from maize.utilities.testing import TestRig
 
@@ -71,7 +71,7 @@ class _MinimiseBase(_BioSimSpaceBase, ABC):
     kcal_per_mol / angstrom**2. Default = 10.0.
     """
 
-    save_name: Parameter[Path] = Parameter(optional=True)
+    save_name: FileParameter[Path] = FileParameter(optional=True)
     """
     If supplied, the output files will be saved with
     this name. E.g., if set to Path("output"), the output

diff --git a/maize/steps/exs/biosimspace/production.py b/maize/steps/exs/biosimspace/production.py
@@ -8,7 +8,7 @@
 
 import pytest
 
-from maize.core.interface import Input, Parameter
+from maize.core.interface import FileParameter, Input, Parameter
 from maize.utilities.testing import TestRig
 
 from ._base import _BioSimSpaceBase
@@ -99,7 +99,7 @@ class _ProductionBase(_BioSimSpaceBase, ABC):
     angstrom**2.
     """
 
-    save_name: Parameter[Path] = Parameter(optional=True)
+    save_name: FileParameter[Path] = FileParameter(optional=True)
     """
     If supplied, the output files will be saved with
     this name. E.g., if set to Path("output"), the output