Merge pull request #597 from roboflow/fix/fix_inference_pipeline

Fix bug with `InferencePipeline` introduced with `VideoMetadata`
roboflow · Aug 22, 2024 · 37623f3 · 37623f3
2 parents 4144d1e + ed0771c
commit 37623f3
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/CODEOWNERS b/CODEOWNERS
@@ -1,2 +1,2 @@
 * @PawelPeczek-Roboflow @grzegorz-roboflow @yeldarby @probicheaux @hansent
-/docs/ @capjamesg
+docs/ @capjamesg
diff --git a/docs/using_inference/inference_pipeline.md b/docs/using_inference/inference_pipeline.md
@@ -187,6 +187,8 @@ pipeline = InferencePipeline.init_with_workflow(
     video_reference="./my_video.mp4",
     workflow_specification=workflow_specification,
     on_prediction=workflows_sink,
+    image_input_name="image",  # adjust according to name of WorkflowImage input you define
+    video_metadata_input_name="video_metadata" # AVAILABLE from v0.17.0! adjust according to name of WorkflowVideoMetadata input you define
 )
 
 # start the pipeline

diff --git a/inference/core/interfaces/stream/inference_pipeline.py b/inference/core/interfaces/stream/inference_pipeline.py
@@ -577,6 +577,7 @@ def init_with_workflow(
                 workflows_parameters=workflows_parameters,
                 execution_engine=execution_engine,
                 image_input_name=image_input_name,
+                video_metadata_input_name=video_metadata_input_name,
             )
         except ImportError as error:
             raise CannotInitialiseModelError(

diff --git a/inference/usage_tracking/collector.py b/inference/usage_tracking/collector.py
@@ -461,7 +461,7 @@ def _flush_queue(self):
         usage_payloads = self._dump_usage_queue_with_lock()
         if not usage_payloads:
             return
-        merged_payloads: APIKeyUsage = self._zip_usage_payloads(
+        merged_payloads: APIKeyUsage = zip_usage_payloads(
             usage_payloads=usage_payloads,
         )
         self._offload_to_api(payloads=merged_payloads)
@@ -477,7 +477,7 @@ def _offload_to_api(self, payloads: List[APIKeyUsage]):
 
         for payload in payloads:
             api_keys_hashes_failed = send_usage_payload(
-                payloads=payloads,
+                payload=payload,
                 api_usage_endpoint_url=self._settings.api_usage_endpoint_url,
                 hashes_to_api_keys=hashes_to_api_keys,
                 ssl_verify=ssl_verify,