Fix: Missing type annotation of Experiment.watch_num_episodes

2024-01-08 18:00:37 +01:00 · 2024-01-08 18:00:37 +01:00 · f77d95da04
commit f77d95da04
parent 97a241a6fc
1 changed files with 1 additions and 1 deletions
--- a/tianshou/highlevel/experiment.py
+++ b/tianshou/highlevel/experiment.py
@ -99,7 +99,7 @@ class ExperimentConfig:
    """Whether to perform training"""
    watch: bool = True
    """Whether to watch agent performance (after training)"""
-    watch_num_episodes = 10
+    watch_num_episodes: int = 10
    """Number of episodes for which to watch performance (if `watch` is enabled)"""
    watch_render: float = 0.0
    """Milliseconds between rendered frames when watching agent performance (if `watch` is enabled)"""