Add scratch of implementation for new Roboflow data collector sink

roboflow · May 17, 2024 · 6abb716 · 6abb716
1 parent d388466
commit 6abb716
Show file tree

Hide file tree

Showing 7 changed files with 385 additions and 67 deletions.
diff --git a/inference/core/roboflow_api.py b/inference/core/roboflow_api.py
@@ -228,7 +228,7 @@ def register_image_at_roboflow(
     image_bytes: bytes,
     batch_name: str,
     tags: Optional[List[str]] = None,
-    inference_id=None,
+    inference_id: Optional[str] = None,
 ) -> dict:
     url = f"{API_BASE_URL}/dataset/{dataset_id}/upload"
     params = [

diff --git a/inference/core/workflows/core_steps/models/roboflow/instance_segmentation.py b/inference/core/workflows/core_steps/models/roboflow/instance_segmentation.py
@@ -141,7 +141,7 @@ class BlockManifest(WorkflowBlockManifest):
     disable_active_learning: Union[
         bool, WorkflowParameterSelector(kind=[BOOLEAN_KIND])
     ] = Field(
-        default=False,
+        default=True,
         description="Parameter to decide if Active Learning data sampling is disabled for the model",
         examples=[True, "$inputs.disable_active_learning"],
     )

diff --git a/inference/core/workflows/core_steps/models/roboflow/keypoint_detection.py b/inference/core/workflows/core_steps/models/roboflow/keypoint_detection.py
@@ -132,7 +132,7 @@ class BlockManifest(WorkflowBlockManifest):
     disable_active_learning: Union[
         bool, WorkflowParameterSelector(kind=[BOOLEAN_KIND])
     ] = Field(
-        default=False,
+        default=True,
         description="Parameter to decide if Active Learning data sampling is disabled for the model",
         examples=[True, "$inputs.disable_active_learning"],
     )

diff --git a/inference/core/workflows/core_steps/models/roboflow/multi_class_classification.py b/inference/core/workflows/core_steps/models/roboflow/multi_class_classification.py
@@ -82,7 +82,7 @@ class BlockManifest(WorkflowBlockManifest):
     disable_active_learning: Union[
         bool, WorkflowParameterSelector(kind=[BOOLEAN_KIND])
     ] = Field(
-        default=False,
+        default=True,
         description="Parameter to decide if Active Learning data sampling is disabled for the model",
         examples=[True, "$inputs.disable_active_learning"],
     )

diff --git a/inference/core/workflows/core_steps/models/roboflow/multi_label_classification.py b/inference/core/workflows/core_steps/models/roboflow/multi_label_classification.py
@@ -84,7 +84,7 @@ class BlockManifest(WorkflowBlockManifest):
     disable_active_learning: Union[
         bool, WorkflowParameterSelector(kind=[BOOLEAN_KIND])
     ] = Field(
-        default=False,
+        default=True,
         description="Parameter to decide if Active Learning data sampling is disabled for the model",
         examples=[True, "$inputs.disable_active_learning"],
     )

diff --git a/inference/core/workflows/core_steps/models/roboflow/object_detection.py b/inference/core/workflows/core_steps/models/roboflow/object_detection.py
@@ -122,7 +122,7 @@ class BlockManifest(WorkflowBlockManifest):
     disable_active_learning: Union[
         bool, WorkflowParameterSelector(kind=[BOOLEAN_KIND])
     ] = Field(
-        default=False,
+        default=True,
         description="Parameter to decide if Active Learning data sampling is disabled for the model",
         examples=[True, "$inputs.disable_active_learning"],
     )