From 2724838cf83ba042dba082de0bcc87976ae3f9fe Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Tue, 30 May 2023 13:14:43 +0100
Subject: [PATCH] Setup testing scripts

---
 src/primaite/config/config_main.yaml |  2 +-
 src/primaite/main.py                 | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/primaite/config/config_main.yaml b/src/primaite/config/config_main.yaml
index b31a73b7..6104e25b 100644
--- a/src/primaite/config/config_main.yaml
+++ b/src/primaite/config/config_main.yaml
@@ -11,7 +11,7 @@ numEpisodes: 10
 # Time delay between steps (for generic agents)
 timeDelay: 10
 # Filename of the scenario / laydown
-configFilename: config_5_DATA_MANIPULATION.yaml
+configFilename: config_1_DDOS_BASIC.yaml
 # Type of session to be run (TRAINING or EVALUATION)
 sessionType: TRAINING
 # Determine whether to load an agent from file
diff --git a/src/primaite/main.py b/src/primaite/main.py
index 0963fa7e..1211b5cd 100644
--- a/src/primaite/main.py
+++ b/src/primaite/main.py
@@ -349,12 +349,12 @@ except Exception:
 transaction_list = []
 
 # Create the Primaite environment
-try:
-    env = Primaite(config_values, transaction_list)
-    logging.info("PrimAITE environment created")
-except Exception:
-    logging.error("Could not create PrimAITE environment")
-    logging.error("Exception occured", exc_info=True)
+# try:
+env = Primaite(config_values, transaction_list)
+#     logging.info("PrimAITE environment created")
+# except Exception:
+#     logging.error("Could not create PrimAITE environment")
+#     logging.error("Exception occured", exc_info=True)
 
 # Get the number of steps (which is stored in the child config file)
 config_values.num_steps = env.episode_steps