lcipolina commited on
Commit
b8eecd0
·
verified ·
1 Parent(s): e3ff159

Fixed parsing of the config files

Browse files
Files changed (1) hide show
  1. scripts/simulate.py +9 -1
scripts/simulate.py CHANGED
@@ -7,11 +7,19 @@ Handles environment creation, policy initialization, and the simulation loop.
7
  """
8
 
9
  import logging
 
 
10
  from typing import Dict, Any
 
 
 
 
 
 
 
11
  from game_reasoning_arena.arena.utils.seeding import set_seed
12
  from game_reasoning_arena.arena.games.registry import registry # Games registry
13
  from game_reasoning_arena.backends import initialize_llm_registry
14
-
15
  from game_reasoning_arena.arena.agents.policy_manager import (
16
  initialize_policies, policy_mapping_fn
17
  )
 
7
  """
8
 
9
  import logging
10
+ import sys
11
+ from pathlib import Path
12
  from typing import Dict, Any
13
+
14
+ # Ensure the src directory is in the Python path
15
+ current_dir = Path(__file__).parent
16
+ src_dir = current_dir / ".." / "src"
17
+ sys.path.insert(0, str(src_dir.resolve()))
18
+
19
+ # pylint: disable=wrong-import-position
20
  from game_reasoning_arena.arena.utils.seeding import set_seed
21
  from game_reasoning_arena.arena.games.registry import registry # Games registry
22
  from game_reasoning_arena.backends import initialize_llm_registry
 
23
  from game_reasoning_arena.arena.agents.policy_manager import (
24
  initialize_policies, policy_mapping_fn
25
  )