From edd64d501dee443f1776ff1e7d6404d0c40703b3 Mon Sep 17 00:00:00 2001
From: LorenzzoQM <lorenzzo.quevedomantovani@colorado.edu>
Date: Thu, 7 Dec 2023 11:24:41 -0700
Subject: [PATCH] Issue #95: Add new_data as a DataStore class attribute
---
src/bsk_rl/envs/general_satellite_tasking/scenario/data.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/src/bsk_rl/envs/general_satellite_tasking/scenario/data.py b/src/bsk_rl/envs/general_satellite_tasking/scenario/data.py
index 19f5ba0a..daa06579 100644
--- 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: