nevernever69 commited on
Commit
e4061c7
·
verified ·
1 Parent(s): 27fb993

fix server/redveil_environment.py for validator

Browse files
Files changed (1) hide show
  1. server/redveil_environment.py +9 -5
server/redveil_environment.py CHANGED
@@ -20,6 +20,10 @@ from uuid import uuid4
20
  from openenv.core.env_server.interfaces import Environment
21
  from openenv.core.env_server.types import State
22
 
 
 
 
 
23
  try:
24
  from ..models import ActionType, RedVeilAction, RedVeilObservation
25
  from ..noise import DeceptionEngine, NoiseEngine
@@ -27,11 +31,11 @@ try:
27
  from ..grader import grade_task
28
  from ..vulnerable_app import create_vulnerable_app
29
  except (ImportError, ModuleNotFoundError):
30
- from models import ActionType, RedVeilAction, RedVeilObservation
31
- from noise import DeceptionEngine, NoiseEngine
32
- from tasks import ALL_TASKS, TaskConfig
33
- from grader import grade_task
34
- from vulnerable_app import create_vulnerable_app
35
 
36
 
37
  # ---------------------------------------------------------------------------
 
20
  from openenv.core.env_server.interfaces import Environment
21
  from openenv.core.env_server.types import State
22
 
23
+ import sys as _sys
24
+ import os as _os
25
+ _sys.path.insert(0, _os.path.dirname(_os.path.dirname(_os.path.abspath(__file__))))
26
+
27
  try:
28
  from ..models import ActionType, RedVeilAction, RedVeilObservation
29
  from ..noise import DeceptionEngine, NoiseEngine
 
31
  from ..grader import grade_task
32
  from ..vulnerable_app import create_vulnerable_app
33
  except (ImportError, ModuleNotFoundError):
34
+ from redveil.models import ActionType, RedVeilAction, RedVeilObservation
35
+ from redveil.noise import DeceptionEngine, NoiseEngine
36
+ from redveil.tasks import ALL_TASKS, TaskConfig
37
+ from redveil.grader import grade_task
38
+ from redveil.vulnerable_app import create_vulnerable_app
39
 
40
 
41
  # ---------------------------------------------------------------------------