#917 - Integrated the PrimaiteSession into all tests.

- Ran a full pre-commit hook and thus encountered tons of fixes required
This commit is contained in:
Chris McCarthy
2023-06-30 09:08:13 +01:00
parent b6d93ad33f
commit 7b1f889415
62 changed files with 1880 additions and 802 deletions

View File

@@ -7,12 +7,7 @@ from typing import List, Tuple
class Transaction(object):
"""Transaction class."""
def __init__(
self,
agent_identifier,
episode_number,
step_number
):
def __init__(self, agent_identifier, episode_number, step_number):
"""
Transaction constructor.
@@ -37,6 +32,11 @@ class Transaction(object):
"The action space invoked by the agent"
def as_csv_data(self) -> Tuple[List, List]:
"""
Converts the Transaction to a csv data row and provides a header.
:return: A tuple consisting of (header, data).
"""
if isinstance(self.action_space, int):
action_length = self.action_space
else:
@@ -74,12 +74,14 @@ class Transaction(object):
str(self.reward),
]
row = (
row
+ _turn_action_space_to_array(self.action_space)
+ _turn_obs_space_to_array(self.obs_space_pre, obs_assets,
obs_features)
+ _turn_obs_space_to_array(self.obs_space_post, obs_assets,
obs_features)
row
+ _turn_action_space_to_array(self.action_space)
+ _turn_obs_space_to_array(
self.obs_space_pre, obs_assets, obs_features
)
+ _turn_obs_space_to_array(
self.obs_space_post, obs_assets, obs_features
)
)
return header, row