Skip to content

Commit d15cf5c

Browse files
committed
Small improves in debugging random attacker
1 parent 70bf197 commit d15cf5c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

agents/attackers/random/random_agent.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,13 @@
1313
sys.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
1717
from base_agent import BaseAgent
1818
from agent_utils import generate_valid_actions
1919
from 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

Comments
 (0)