From 8a24427bf79e5625d6c597b66b6619c7d846a651 Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Fri, 26 May 2023 14:50:15 +0100 Subject: [PATCH] Fix minor logic errors in main script --- src/primaite/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/primaite/main.py b/src/primaite/main.py index 0963fa7e..0f94c3f8 100644 --- a/src/primaite/main.py +++ b/src/primaite/main.py @@ -25,6 +25,7 @@ from primaite.transactions.transactions_to_file import write_transaction_to_file def run_generic(): """Run against a generic agent.""" for episode in range(0, config_values.num_episodes): + env.reset() for step in range(0, config_values.num_steps): # Send the observation space to the agent to get an action # TEMP - random action for now @@ -42,7 +43,6 @@ def run_generic(): time.sleep(config_values.time_delay / 1000) # Reset the environment at the end of the episode - env.reset() env.close() @@ -375,7 +375,7 @@ logging.info("Saving transaction logs...") write_transaction_to_file(transaction_list) -config_file_main.close +config_file_main.close() print("Finished") logging.info("Finished")