Upload strategies/data_recorder.py with huggingface_hub

strategies/data_recorder.py

@@ -210,25 +210,35 @@ class DataRecorderStrategy(Strategy):
     def _test_connectivity(self):
         """
         Attempts to upload a small hello_world file to verify HF token permissions.
+        Retries 3 times to handle startup network lag.
         """
-        try:
-            self.log.info("Testing HuggingFace Dataset connectivity...", LogColor.YELLOW)
-            test_file = Path("connectivity_test.txt")
-            test_file.write_text(f"Connected at {datetime.utcnow()}")
-            
-            self.api.upload_file(
-                path_or_fileobj=str(test_file),
-                path_in_repo="connectivity_test.txt",
-                repo_id=self.repo_id,
-                repo_type="dataset",
-            )
-            self.log.info("✅ Connectivity Test PASSED! Token is valid.", LogColor.GREEN)
-            test_file.unlink() # Cleanup
-        except Exception as e:
-            self.log.error(f"❌ Connectivity Test FAILED: {e}", LogColor.RED)
-            self.log.error("Please check HF_TOKEN in Spaces Secrets.", LogColor.RED)
+        max_retries = 3
+        for i in range(max_retries):
+            try:
+                self.log.info(f"Testing HuggingFace Dataset connectivity (Attempt {i+1}/{max_retries})...", LogColor.YELLOW)
+                test_file = Path("connectivity_test.txt")
+                test_file.write_text(f"Connected at {datetime.utcnow()}")
+                
+                self.api.upload_file(
+                    path_or_fileobj=str(test_file),
+                    path_in_repo="connectivity_test.txt",
+                    repo_id=self.repo_id,
+                    repo_type="dataset",
+                )
+                self.log.info("✅ Connectivity Test PASSED! Token is valid.", LogColor.GREEN)
+                test_file.unlink() # Cleanup
+                break
+            except Exception as e:
+                self.log.error(f"⚠️ Connectivity Attempt {i+1} Failed: {e}", LogColor.YELLOW)
+                if i < max_retries - 1:
+                    time.sleep(5)
+                else:
+                    self.log.error(f"❌ Connectivity Test FAILED after {max_retries} attempts.", LogColor.RED)
+                    self.log.error("Please check HF_TOKEN in Spaces Secrets.", LogColor.RED)
+        
         if self.catalog_path.exists():
             shutil.rmtree(self.catalog_path)
+
         self.catalog_path.mkdir(parents=True, exist_ok=True)
 
     # -- EVENT HANDLERS ----------------------------------------------------------------