From dd8809b51713c6ef3841c6abe7233a805b507fc8 Mon Sep 17 00:00:00 2001 From: sp Date: Sat, 30 Dec 2023 13:35:55 +0100 Subject: [PATCH] trying add_text --- examples/shields/rl/callbacks.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/shields/rl/callbacks.py b/examples/shields/rl/callbacks.py index 03c724d..9fe1e24 100644 --- a/examples/shields/rl/callbacks.py +++ b/examples/shields/rl/callbacks.py @@ -35,9 +35,8 @@ 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(f"{worker.io_context.log_dir}/shield_data") - print(file_writer) - with file_writer.as_default(): - tf.summary.text("first_text_from_episode_start", "testing_in_episode", step=0) + print(file_writer.logdir) + file_writer.add_text("first_text_from_episode_start", "testing_in_episode", 0) file_writer.flush() # print(F"Epsiode started Environment: {base_env.get_sub_environments()}") env = base_env.get_sub_environments()[0]