Black

HumanSignal · Apr 29, 2024 · 334d476 · 334d476
1 parent 7bee943
commit 334d476
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/label_studio_sdk/project.py b/label_studio_sdk/project.py
@@ -1124,10 +1124,10 @@ def create_prediction(
         """
         data = {"task": task_id, "result": result, "score": score}
         if model_version is not None:
-            data['model_version'] = model_version
-        response = self.make_request('POST', "/api/predictions", json=data)
+            data["model_version"] = model_version
+        response = self.make_request("POST", "/api/predictions", json=data)
         json = response.json()
-        logger.debug(f'Response: {json}')
+        logger.debug(f"Response: {json}")
         return json
 
     def create_predictions(self, predictions):
@@ -2324,8 +2324,8 @@ def export(
         self,
         filters=None,
         title="SDK Export",
-        export_type='JSON',
-        output_dir='.',
+        export_type="JSON",
+        output_dir=".",
         **kwargs,
     ):
         """
@@ -2385,8 +2385,8 @@ def export(
 
         # Create a temporary view with the specified filters
         if filters:
-            view = self.create_view(title='Temp SDK export', filters=filters)
-            task_filter_options = {'view': view['id']}
+            view = self.create_view(title="Temp SDK export", filters=filters)
+            task_filter_options = {"view": view["id"]}
         else:
             task_filter_options = None
             view = None
@@ -2399,7 +2399,7 @@ def export(
         )
 
         # Check the status of the snapshot creation
-        export_id = export_result['id']
+        export_id = export_result["id"]
         while self.export_snapshot_status(export_id).is_in_progress():
             time.sleep(1.0)  # Wait until the snapshot is ready
 
@@ -2412,8 +2412,8 @@ def export(
 
         # Clean up the view
         if view:
-            self.delete_view(view['id'])
-        return {'status': status, 'filename': filename, 'export_id': export_id}
+            self.delete_view(view["id"])
+        return {"status": status, "filename": filename, "export_id": export_id}
 
     def export_snapshot_status(self, export_id: int) -> ExportSnapshotStatus:
         """