chore: cleanup datasets and robots in record_and_playback integration

neuracore/api/endpoints.py

@@ -6,9 +6,14 @@
 """
 
 import requests
-from neuracore_types import DataSpec, DataType, DeploymentRequest, SynchronizedPoint
+from neuracore_types import (
+    DataSpec,
+    DataType,
+    DeploymentRequest,
+    GPUType,
+    SynchronizedPoint,
+)
 from neuracore_types.endpoints.endpoint_requests import DeploymentConfig
-from neuracore_types.training.training import GPUType
 
 from neuracore.api.core import _get_robot
 from neuracore.core.auth import get_auth

neuracore/core/robot.py

@@ -701,6 +701,24 @@ def _cleanup_daemon_recording_context(self) -> None:
         finally:
             self._daemon_recording_context = None
 
+    def delete(self) -> None:
+        """Delete the robot from the server."""
+        try:
+            response = requests.delete(
+                f"{API_URL}/org/{self.org_id}/robots/{self.id}",
+                headers=self._auth.get_headers(),
+            )
+            response.raise_for_status()
+        except requests.exceptions.ConnectionError:
+            raise RobotError(
+                "Failed to connect to neuracore server, "
+                "please check your internet connection and try again."
+            )
+        except requests.exceptions.RequestException as e:
+            raise RobotError(f"Failed to delete robot: {str(e)}")
+        finally:
+            self.close()
+
     def close(self) -> None:
         """Release local resources owned by this Robot instance."""
         self._cleanup_daemon_recording_context()

tests/integration/ml/test_algorithm_performance.py

@@ -33,11 +33,11 @@
 from neuracore_types import (
     BatchedJointData,
     DataType,
+    GPUType,
     JointData,
     RGBCameraData,
     SynchronizedPoint,
 )
-from neuracore_types.training.training import GPUType
 
 import neuracore as nc
 from neuracore.core.endpoint import Policy