Sushruth21 commited on
Commit
cf4926b
·
1 Parent(s): be60b79

fix: update validation scripts with proper he_demo imports for openenv validation

Browse files
Files changed (2) hide show
  1. validate.py +3 -26
  2. validate_comprehensive.py +3 -17
validate.py CHANGED
@@ -5,32 +5,9 @@ Final validation script for the Energy & Memory RAM Optimization Environment.
5
  Per hackathon requirements, this validation includes grader configuration verification.
6
  """
7
 
8
- import sys
9
- import os
10
-
11
- # Add the project root to Python path
12
- project_root = os.path.dirname(__file__)
13
- sys.path.insert(0, project_root)
14
-
15
- # Mock the he_demo package FIRST before any imports
16
- import types
17
- he_demo = types.ModuleType('he_demo')
18
-
19
- # Import models and graders and add to he_demo
20
- from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
21
- from task_graders import TASK_GRADERS, get_grader_metadata, get_grader
22
- he_demo.EnergyOptimizationAction = EnergyOptimizationAction
23
- he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
24
- he_demo.Task = Task
25
- he_demo.TaskSummary = TaskSummary
26
- he_demo.task_graders = sys.modules.get('task_graders')
27
-
28
- # Add to sys.modules
29
- sys.modules['he_demo'] = he_demo
30
- sys.modules['he_demo.models'] = he_demo
31
- sys.modules['he_demo.task_graders'] = sys.modules.get('task_graders')
32
-
33
- # Now import the environment
34
  from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
35
 
36
  def main():
 
5
  Per hackathon requirements, this validation includes grader configuration verification.
6
  """
7
 
8
+ # Import from properly structured he_demo package
9
+ from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
10
+ from he_demo.task_graders import TASK_GRADERS, get_grader_metadata, get_grader
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
12
 
13
  def main():
validate_comprehensive.py CHANGED
@@ -11,23 +11,9 @@ import os
11
  project_root = os.path.dirname(__file__)
12
  sys.path.insert(0, project_root)
13
 
14
- # Mock the he_demo package for testing
15
- import types
16
- he_demo = types.ModuleType('he_demo')
17
-
18
- # Import models and add to he_demo
19
- from models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
20
- from task_graders import TASK_GRADERS, get_grader, get_grader_metadata
21
- he_demo.EnergyOptimizationAction = EnergyOptimizationAction
22
- he_demo.EnergyOptimizationObservation = EnergyOptimizationObservation
23
- he_demo.Task = Task
24
- he_demo.TaskSummary = TaskSummary
25
-
26
- # Add to sys.modules
27
- sys.modules['he_demo'] = he_demo
28
- sys.modules['he_demo.models'] = he_demo
29
-
30
- # Now import the environment
31
  from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
32
 
33
  def create_observation(ram_usage, energy_consumption, steps_taken):
 
11
  project_root = os.path.dirname(__file__)
12
  sys.path.insert(0, project_root)
13
 
14
+ # Import from properly structured he_demo package
15
+ from he_demo.models import EnergyOptimizationAction, EnergyOptimizationObservation, Task, TaskSummary
16
+ from he_demo.task_graders import TASK_GRADERS, get_grader, get_grader_metadata
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  from he_demo.server.he_demo_environment import EnergyOptimizationEnvironment
18
 
19
  def create_observation(ram_usage, energy_consumption, steps_taken):