samrat-rm commited on
Commit
04666da
·
1 Parent(s): a0518e7

chore: import statement refactor

Browse files
Files changed (1) hide show
  1. server/WhyDidItFail_environment.py +3 -7
server/WhyDidItFail_environment.py CHANGED
@@ -13,13 +13,9 @@ from uuid import uuid4
13
  from openenv.core.env_server.interfaces import Environment
14
  from openenv.core.env_server.types import State
15
 
16
- try:
17
- from ..models import WhyDidItFailAction, WhyDidItFailObservation
18
- from ..server.scenarios import SCENARIOS
19
- except ImportError:
20
- from models import WhyDidItFailAction, WhyDidItFailObservation
21
- from server.scenarios import SCENARIOS
22
-
23
 
24
  class WhyDidItFailEnvironment(Environment):
25
  """Diagnostic environment where the agent investigates a failed training run."""
 
13
  from openenv.core.env_server.interfaces import Environment
14
  from openenv.core.env_server.types import State
15
 
16
+ from models import WhyDidItFailAction, WhyDidItFailObservation
17
+ from server.scenarios import SCENARIOS
18
+ from server.graders import grade_easy
 
 
 
 
19
 
20
  class WhyDidItFailEnvironment(Environment):
21
  """Diagnostic environment where the agent investigates a failed training run."""