From 1d6b12ba5aa0337daf63d5b494ae3a6f0ef5e67e Mon Sep 17 00:00:00 2001 From: sp Date: Sat, 30 Dec 2023 12:42:21 +0100 Subject: [PATCH] fixed typo --- examples/shields/rl/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/shields/rl/callbacks.py b/examples/shields/rl/callbacks.py index 0f16b87..4270d15 100644 --- a/examples/shields/rl/callbacks.py +++ b/examples/shields/rl/callbacks.py @@ -34,7 +34,7 @@ class MyCallbacks(DefaultCallbacks): def on_episode_start(self, *, worker: RolloutWorker, base_env: BaseEnv, policies: Dict[PolicyID, Policy], episode, env_index, **kwargs) -> None: file_writer = tf.summary.create_file_writer(worker.io_context.log_dir) - with self.file_writer.as_default(): + with file_writer.as_default(): tf.summary.text("first_text_from_episode_start", "testing_in_episode", step=0) # print(F"Epsiode started Environment: {base_env.get_sub_environments()}") env = base_env.get_sub_environments()[0]