Skip to content

Commit

Permalink
Merge pull request #789 from RossLote/main
Browse files Browse the repository at this point in the history
Allow easier inheritance of pipeline
  • Loading branch information
PawelPeczek-Roboflow authored Nov 8, 2024
2 parents 64a8f46 + 0dca7d1 commit 2feedf7
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions inference/core/interfaces/stream/inference_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,7 @@ def init(
on_prediction = partial(multi_sink, sinks=[on_prediction, al_sink])
on_pipeline_start = active_learning_middleware.start_registration_thread
on_pipeline_end = active_learning_middleware.stop_registration_thread
return InferencePipeline.init_with_custom_logic(
return cls.init_with_custom_logic(
video_reference=video_reference,
on_video_frame=on_video_frame,
on_prediction=on_prediction,
Expand Down Expand Up @@ -422,7 +422,7 @@ def init_with_yolo_world(
f"Could not initialise yolo_world/{model_size} due to lack of sufficient dependencies. "
f"Use pip install inference[yolo-world] to install missing dependencies and try again."
) from error
return InferencePipeline.init_with_custom_logic(
return cls.init_with_custom_logic(
video_reference=video_reference,
on_video_frame=on_video_frame,
on_prediction=on_prediction,
Expand Down Expand Up @@ -642,7 +642,7 @@ def init_with_workflow(
profiler=profiler,
profiling_directory=profiling_directory,
)
return InferencePipeline.init_with_custom_logic(
return cls.init_with_custom_logic(
video_reference=video_reference,
on_video_frame=on_video_frame,
on_prediction=on_prediction,
Expand Down

0 comments on commit 2feedf7

Please sign in to comment.