Add missing definition of sink variable

nv-morpheus · Nov 8, 2023 · ff328a2 · ff328a2
1 parent 16c6ca5
commit ff328a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/llm/agents/kafka_pipeline.py b/examples/llm/agents/kafka_pipeline.py
@@ -86,7 +86,7 @@ def pipeline(num_threads, pipeline_batch_size, model_max_batch_size, model_name)
 
     pipe.add_stage(LLMEngineStage(config, engine=_build_engine(model_name=model_name)))
 
-    pipe.add_stage(InMemorySinkStage(config))
+    sink = pipe.add_stage(InMemorySinkStage(config))
 
     # pipe.add_stage(MonitorStage(config, description="Upload rate", unit="events", delayed_start=True))