From f77d95da0485fc6f55b957632ddc66dc56a5563e Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Mon, 8 Jan 2024 18:00:37 +0100 Subject: [PATCH] Fix: Missing type annotation of Experiment.watch_num_episodes --- tianshou/highlevel/experiment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tianshou/highlevel/experiment.py b/tianshou/highlevel/experiment.py index 3989d35..6bf8e41 100644 --- 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)"""