Skip to content

Commit 055b064

Browse files
Add type parameter to configuration variable
1 parent 2d473e5 commit 055b064

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causal_testing/surrogate/causal_surrogate_assisted.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,10 +53,10 @@ def startup(self, **kwargs):
5353
def shutdown(self, **kwargs):
5454
"""Function to safely exit and shutdown the Simulator"""
5555

56-
def run_with_config(self, configuration) -> SimulationResult:
56+
def run_with_config(self, configuration: Any) -> SimulationResult:
5757
"""Run the simulator with the given configuration and return the results in the structure of a
5858
SimulationResult
59-
:param configuration: the configuration required to initialise the Simulation
59+
:param configuration: The configuration required to initialise the Simulation
6060
:return: Simulation results in the structure of the SimulationResult data class"""
6161

6262

0 commit comments

Comments
 (0)