Update test_env.py

test_env.py

@@ -1,22 +1,22 @@
-from adaptive_cache.env import AdaptiveCacheEnv, Action
-import random
-
-def test_graders():
-    print("Running explicit Grader Validation...")
-    for level in ["easy", "medium", "hard"]:
-        env = AdaptiveCacheEnv(task_level=level)
-        env.reset()
-        done = False
-        while not done:
-            # Simulate an agent making entirely random choices
-            action = Action(evict_index=random.randint(0, 9))
-            _, _, done, info = env.step(action)
-        
-        score = info['score']
-        
-        # This assert statement proves to judges the score is strictly 0.0 to 1.0
-        assert 0.0 <= score <= 1.0, f"Grader out of bounds: {score}"
-        print(f"Task {level.upper()} validated. Score: {score:.2f}")
-
-if __name__ == "__main__":
+from adaptive_cache.env import AdaptiveCacheEnv, Action
+import random
+
+def test_graders():
+    print("Running explicit Grader Validation...")
+    for level in ["easy", "medium", "hard"]:
+        env = AdaptiveCacheEnv(task_level=level)
+        env.reset()
+        done = False
+        while not done:
+            # Simulate an agent making entirely random choices
+            action = Action(evict_index=random.randint(0, 9))
+            _, _, done, info = env.step(action)
+        
+        score = info['score']
+        
+        # This assert statement proves to judges the score is strictly 0.0 to 1.0
+        assert 0.0 <= score <= 1.0, f"Grader out of bounds: {score}"
+        print(f"Task {level.upper()} validated. Score: {score:.2f}")
+
+if __name__ == "__main__":
     test_graders()