1313sys .path .append (os .path .dirname (os .path .dirname (os .path .dirname (os .path .abspath (__file__ ) ))))
1414
1515# with the path fixed, we can import now
16- from env .game_components import Action , Observation , GameState
16+ from env .game_components import Action , Observation
1717from base_agent import BaseAgent
1818from agent_utils import generate_valid_actions
1919from datetime import datetime
2020
2121
22- class RandomAgent (BaseAgent ):
22+ class RandomAttackerAgent (BaseAgent ):
2323
2424 def __init__ (self , host , port ,role , seed ) -> None :
2525 super ().__init__ (host , port , role )
@@ -75,7 +75,7 @@ def select_action(self, observation:Observation)->Action:
7575 logging .basicConfig (filename = os .path .join (args .logdir , "random_agent.log" ), filemode = 'w' , format = '%(asctime)s %(name)s %(levelname)s %(message)s' , datefmt = '%H:%M:%S' ,level = logging .INFO )
7676
7777 # Create agent
78- agent = RandomAgent (args .host , args .port ,"Attacker" , seed = 42 )
78+ agent = RandomAttackerAgent (args .host , args .port ,"Attacker" , seed = 42 )
7979
8080 if not args .evaluate :
8181 # Play the normal game
@@ -94,7 +94,7 @@ def select_action(self, observation:Observation)->Action:
9494 # - At the end, report in log and mlflow and console
9595
9696 # Mlflow experiment name
97- experiment_name = "Evaluation of Random Agent"
97+ experiment_name = "Evaluation of Random Attacker Agent"
9898 if args .mlflow_url :
9999 mlflow .set_tracking_uri (args .mlflow_url )
100100 mlflow .set_experiment (experiment_name )
0 commit comments