Merge branch 'main' into cyst-integration

stratosphereips · Jan 27, 2025 · 77e5edd · 77e5edd
2 parents 7f6f13f + d15cf5c
commit 77e5edd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/agents/attackers/random/random_agent.py b/agents/attackers/random/random_agent.py
@@ -10,6 +10,7 @@
 from AIDojoCoordinator.game_components import Action, Observation, AgentStatus
 
 sys.path.append(path.dirname(path.dirname(path.dirname(path.abspath(__file__) ))))
+
 # with the path fixed, we can import now
 from base_agent import BaseAgent
 from agent_utils import generate_valid_actions
@@ -62,14 +63,14 @@ def select_action(self, observation:Observation)->Action:
     parser.add_argument("--port", help="Port where the game server is", default=9000, type=int, action='store', required=False)
     parser.add_argument("--episodes", help="Sets number of episodes to play or evaluate", default=100, type=int) 
     parser.add_argument("--test_each", help="Evaluate performance during testing every this number of episodes.", default=10, type=int)
-    parser.add_argument("--logdir", help="Folder to store logs", default=path.join(path.dirname(path.abspath(__file__)), "logs"))
+    parser.add_argument("--logdir", help="Folder to store logs", default=os.path.join(os.path.dirname(os.path.abspath(__file__)), "logs"))
     parser.add_argument("--evaluate", help="Evaluate the agent and report, instead of playing the game only once.", default=True)
     parser.add_argument("--mlflow_url", help="URL for mlflow tracking server. If not provided, mlflow will store locally.", default=None)
     args = parser.parse_args()
 
-    if not path.exists(args.logdir):
-        makedirs(args.logdir)
-    logging.basicConfig(filename=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)
+    if not os.path.exists(args.logdir):
+        os.makedirs(args.logdir)
+    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)
 
     # Create agent
     agent = RandomAttackerAgent(args.host, args.port,"Attacker", seed=42)

diff --git a/agents/defenders/random/random_agent.py b/agents/defenders/random/random_agent.py
@@ -11,10 +11,7 @@
 import mlflow
 from os import path
 
-# This is used so the agent can see the environment and game components
-sys.path.append(
-    path.dirname(path.dirname(path.dirname(path.dirname(path.dirname(path.abspath(__file__))))))
-)
+
 # importing agent utils and base agent
 sys.path.append(path.dirname(path.dirname(path.dirname(path.abspath(__file__) ))))
 # with the path fixed, we can import now