Merge remote-tracking branch 'origin/dev' into feature/917_Integrate_with_RLLib
# Conflicts: # src/primaite/config/_package_data/training/training_config_main.yaml # src/primaite/environment/primaite_env.py # src/primaite/main.py # src/primaite/transactions/transaction.py # src/primaite/transactions/transactions_to_file.py
This commit is contained in:
@@ -31,6 +31,16 @@ class Transaction(object):
|
||||
self.action_space = None
|
||||
"The action space invoked by the agent"
|
||||
|
||||
def set_obs_space(self, _obs_space):
|
||||
"""
|
||||
Sets the observation space (pre).
|
||||
|
||||
Args:
|
||||
_obs_space_pre: The observation space before any actions are taken
|
||||
"""
|
||||
self.obs_space = _obs_space
|
||||
|
||||
|
||||
def as_csv_data(self) -> Tuple[List, List]:
|
||||
"""
|
||||
Converts the Transaction to a csv data row and provides a header.
|
||||
|
||||
Reference in New Issue
Block a user