diff --git a/src/pybamm/experiment/experiment.py b/src/pybamm/experiment/experiment.py index fb20a0180e..ce44457cb2 100644 --- a/src/pybamm/experiment/experiment.py +++ b/src/pybamm/experiment/experiment.py @@ -3,6 +3,7 @@ from .step.base_step import ( _convert_time_to_seconds, _convert_temperature_to_kelvin, + BaseStep, ) @@ -39,7 +40,7 @@ class Experiment: def __init__( self, - operating_conditions: list[str | tuple[str]], + operating_conditions: list[str | tuple[str] | BaseStep], period: str | None = None, temperature: float | None = None, termination: list[str] | None = None,