Upload env.py

env.py

@@ -43,11 +43,8 @@ def calculate_iou(box1: List[List[int]], box2: List[List[int]]) -> float:
     return inter_area / union_area
 
 class MetaOCTEnv:
-    def __init__(self, data_dir: str = "images", truth_file: str = "ground_truth.json", difficulty: str = "medium"):
+    def __init__(self, data_dir: str = ".", truth_file: str = "ground_truth.json", difficulty: str = "medium"):
         self.data_dir = data_dir
-        # Support both flat root upload (HuggingFace) and local images/ folder
-        if not os.path.isdir(data_dir):
-            self.data_dir = "."  # Fallback to root directory
         with open(truth_file, "r") as f:
             self.ground_truth = json.load(f)
         self.image_files = list(self.ground_truth.keys())