Setup testing scripts

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

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