Ignore mypy

araffin · araffin · commit 8731e201e2ae · 2023-11-08T11:07:33.000+01:00
diff --git a/rl_zoo3/exp_manager.py b/rl_zoo3/exp_manager.py
@@ -789,7 +789,7 @@ def objective(self, trial: optuna.Trial) -> float:
             )
 
         try:
-            model.learn(self.n_timesteps, callback=callbacks, **learn_kwargs)
+            model.learn(self.n_timesteps, callback=callbacks, **learn_kwargs)  # type: ignore[arg-type]
             # Free memory
             assert model.env is not None
             model.env.close()

Original file line number	Diff line number	Diff line change
`@@ -789,7 +789,7 @@ def objective(self, trial: optuna.Trial) -> float:`
`789`	`789`	`)`
`790`	`790`
`791`	`791`	`try:`
`792`		`- model.learn(self.n_timesteps, callback=callbacks, **learn_kwargs)`
	`792`	`+ model.learn(self.n_timesteps, callback=callbacks, **learn_kwargs) # type: ignore[arg-type]`
`793`	`793`	`# Free memory`
`794`	`794`	`assert model.env is not None`
`795`	`795`	`model.env.close()`