bugfix

Limmen · Feb 12, 2024 · e6e7e3d · e6e7e3d
1 parent 3960700
commit e6e7e3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/simulation-system/libs/csle-agents/src/csle_agents/agents/ppo/ppo_agent.py b/simulation-system/libs/csle-agents/src/csle_agents/agents/ppo/ppo_agent.py
@@ -206,7 +206,7 @@ def train(self) -> ExperimentExecution:
                 os.chmod(save_path, 0o777)
 
             # Save latest trace
-            if self.save_to_metastore:
+            if self.save_to_metastore and len(orig_env.get_traces()) > 0:
                 MetastoreFacade.save_simulation_trace(orig_env.get_traces()[-1])
             orig_env.reset_traces()