oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/dev' into feature/1468-observations-class

Browse Source
This commit is contained in:
Marek Wolan
2023-06-07 14:39:20 +01:00
parent c969bc32f5 ef3cef530b
commit 89cea9289b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/primaite/main.py
View File

@@ -25,6 +25,7 @@ from primaite.transactions.transactions_to_file import write_transaction_to_file
def run_generic(): def run_generic():
"""Run against a generic agent.""" """Run against a generic agent."""
for episode in range(0, config_values.num_episodes): for episode in range(0, config_values.num_episodes):
env.reset()
for step in range(0, config_values.num_steps): for step in range(0, config_values.num_steps):
# Send the observation space to the agent to get an action # Send the observation space to the agent to get an action
# TEMP - random action for now # TEMP - random action for now
@@ -42,7 +43,6 @@ def run_generic():
time.sleep(config_values.time_delay / 1000) time.sleep(config_values.time_delay / 1000)
# Reset the environment at the end of the episode # Reset the environment at the end of the episode
env.reset()
env.close() env.close()
@@ -375,7 +375,7 @@ logging.info("Saving transaction logs...")
write_transaction_to_file(transaction_list) write_transaction_to_file(transaction_list)
config_file_main.close config_file_main.close()
print("Finished") print("Finished")
logging.info("Finished") logging.info("Finished")