Browse Source

fixed typo

refactoring
sp 11 months ago
parent
commit
1d6b12ba5a
  1. 2
      examples/shields/rl/callbacks.py

2
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: 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) 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) tf.summary.text("first_text_from_episode_start", "testing_in_episode", step=0)
# print(F"Epsiode started Environment: {base_env.get_sub_environments()}") # print(F"Epsiode started Environment: {base_env.get_sub_environments()}")
env = base_env.get_sub_environments()[0] env = base_env.get_sub_environments()[0]

Loading…
Cancel
Save