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

Setup testing scripts

Browse Source
This commit is contained in:
Marek Wolan
2023-05-30 13:14:43 +01:00
parent b255f557db
commit 2724838cf8
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/config/config_main.yaml
View File

@@ -11,7 +11,7 @@ numEpisodes: 10
# Time delay between steps (for generic agents) # Time delay between steps (for generic agents)
timeDelay: 10 timeDelay: 10
# Filename of the scenario / laydown # 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) # Type of session to be run (TRAINING or EVALUATION)
sessionType: TRAINING sessionType: TRAINING
# Determine whether to load an agent from file # Determine whether to load an agent from file

12
src/primaite/main.py
View File

@@ -349,12 +349,12 @@ except Exception:
transaction_list = [] transaction_list = []
# Create the Primaite environment # Create the Primaite environment
try: # try:
env = Primaite(config_values, transaction_list) env = Primaite(config_values, transaction_list)
logging.info("PrimAITE environment created") # logging.info("PrimAITE environment created")
except Exception: # except Exception:
logging.error("Could not create PrimAITE environment") # logging.error("Could not create PrimAITE environment")
logging.error("Exception occured", exc_info=True) # logging.error("Exception occured", exc_info=True)
# Get the number of steps (which is stored in the child config file) # Get the number of steps (which is stored in the child config file)
config_values.num_steps = env.episode_steps config_values.num_steps = env.episode_steps