finalise the fix RunnableSimulationWithMemories pytype fix.

PiperOrigin-RevId: 675573771 Change-Id: I64e7536f64206edb26aaacbb91600e09239d704d
google-deepmind · Sep 17, 2024 · b65862a · b65862a
1 parent 3d5ae49
commit b65862a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/modular/scenario/scenarios.py b/examples/modular/scenario/scenarios.py
@@ -299,7 +299,7 @@ def build_simulation(
     agent_base_module: str = DEFAULT_IMPORT_AGENT_BASE_MODULE,
     support_agent_base_module: str = DEFAULT_IMPORT_SUPPORT_AGENT_MODULE,
     env_base_module: str = DEFAULT_IMPORT_ENV_BASE_MODULE,
-) -> Runnable:
+) -> RunnableSimulationWithMemories:
   """Builds a simulation from a scenario configuration."""
   substrate_config = scenario_config.substrate_config
   # Load the environment config with importlib