Issue #95: Add new_data as a DataStore class attribute

AVSLab · Dec 7, 2023 · 68839e6 · 68839e6
1 parent ca37e48
commit 68839e6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/bsk_rl/envs/general_satellite_tasking/scenario/data.py b/src/bsk_rl/envs/general_satellite_tasking/scenario/data.py
@@ -40,6 +40,7 @@ def __init__(self, data_manager: "DataManager", satellite: "Satellite") -> None:
 
         self._initialize_knowledge(data_manager.env_features)
         self.data = self.DataType()
+        self.new_data = self.DataType()
 
     def _initialize_knowledge(self, env_features: "EnvironmentFeatures") -> None:
         """Establish knowledge about the world known to the satellite. Defaults to
@@ -84,6 +85,7 @@ def internal_update(self) -> "DataType":
         self._clear_logs()
         new_data = self._compare_log_states(old_log_state, self.log_state)
         self.data += new_data
+        self.new_data = new_data
         return new_data
 
     def stage_communicated_data(self, external_data: "DataType") -> None: