Add files

delete_episodes.py

@@ -54,7 +54,7 @@ def download_dataset(
 
 
 def check_v2_format(dataset_path: str) -> bool:
-    """Check if dataset is in v2.0 format"""
+    """Check if dataset is in v2.x format"""
     info_path = os.path.join(dataset_path, "meta", "info.json")
     
     if not os.path.exists(info_path):
@@ -64,11 +64,16 @@ def check_v2_format(dataset_path: str) -> bool:
         try:
             info = json.load(f)
             if "codebase_version" not in info:
-                raise ValueError(f"Error: {info_path} is not a valid v2.0 dataset")
-            elif info["codebase_version"] != "v2.0":
+                raise ValueError(f"Error: {info_path} is not a valid v2.x dataset")
+            
+            version = info["codebase_version"]
+            # Accept any v2.x version (v2.0, v2.1, etc.)
+            if not version.startswith("v2."):
                 raise ValueError(
-                    f"Error: {info_path} is not a v2.0 dataset, found {info['codebase_version']}"
+                    f"Error: {info_path} is not a v2.x dataset, found {version}"
                 )
+            
+            logger.info(f"Dataset version: {version}")
             return True
         except json.JSONDecodeError:
             raise ValueError(f"Error: {info_path} is not a valid JSON file")