class_docstr: Simple setup for logging of environment events with all their attributes.
  
  Events will be logged into CSV files in "event_logs" directory.
  
  :param env: Env to log events from (needs to be wrapper in LogEventsWrapper)
class_docstr_tok: ["Simple","setup","for","logging","of","environment","events","with","all","their","attributes",".","Events","will","be","logged","into","CSV","files","in","``","event_logs","''","directory",".",":","param","env",":","Env","to","log","events","from","(","needs","to","be","wrapper","in","LogEventsWrapper",")"]
docstr: Ensure the event logs are written out in case the rollout ended in the middle of episode
  (without env reset at the end).
docstr_tok: ["Ensure","the","event","logs","are","written","out","in","case","the","rollout","ended","in","the","middle","of","episode","(","without","env","reset","at","the","end",")","."]
code: def __exit__(self, exc_type, exc_val, exc_tb) -> None:
      """
      Ensure the event logs are written out in case the rollout ended in the middle of episode
      (without env reset at the end).
      """
      self.env.write_epoch_stats()
  
imports: from maze.core.log_events.log_events_writer_registry import LogEventsWriterRegistry
  from maze.core.log_events.log_events_writer_tsv import LogEventsWriterTSV
  from maze.core.wrappers.log_stats_wrapper import LogStatsWrapper
label_desc: Use image node_id 3 for calling the SimpleEventLoggingSetup obj's underlying member method code with example usage: obj.__exit__(exc_type, exc_val, exc_tb) without return types