Skip to content

Commit b743c1e

Browse files
committed
feat: Keep event type as complete instead of other
Signed-off-by: ntkathole <nikhilkathole2683@gmail.com>
1 parent 692ad9e commit b743c1e

File tree

2 files changed

+15
-11
lines changed

2 files changed

+15
-11
lines changed

sdk/python/feast/openlineage/emitter.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -217,11 +217,11 @@ def emit_feature_view_lineage(
217217
namespace=namespace,
218218
)
219219

220-
# Emit a RunEvent with OTHER state (metadata registration, not execution)
220+
# Emit a RunEvent with COMPLETE state to create lineage connection
221221
result = self._client.emit_run_event(
222222
job_name=job.name,
223223
run_id=str(uuid.uuid4()),
224-
event_type=RunState.OTHER,
224+
event_type=RunState.COMPLETE,
225225
inputs=inputs,
226226
outputs=outputs,
227227
job_facets=job.facets,
@@ -262,11 +262,11 @@ def emit_stream_feature_view_lineage(
262262
namespace=namespace,
263263
)
264264

265-
# Emit a RunEvent with OTHER state (metadata registration, not execution)
265+
# Emit a RunEvent with COMPLETE state to create lineage connection
266266
return self._client.emit_run_event(
267267
job_name=f"stream_{job.name}",
268268
run_id=str(uuid.uuid4()),
269-
event_type=RunState.OTHER,
269+
event_type=RunState.COMPLETE,
270270
inputs=inputs,
271271
outputs=outputs,
272272
job_facets=job.facets,
@@ -353,11 +353,11 @@ def emit_on_demand_feature_view_lineage(
353353
)
354354
}
355355

356-
# Emit a RunEvent with OTHER state (metadata registration, not execution)
356+
# Emit a RunEvent with COMPLETE state to create lineage connection
357357
return self._client.emit_run_event(
358358
job_name=f"on_demand_feature_view_{odfv.name}",
359359
run_id=str(uuid.uuid4()),
360-
event_type=RunState.OTHER,
360+
event_type=RunState.COMPLETE,
361361
inputs=inputs,
362362
outputs=outputs,
363363
job_facets=job_facets,
@@ -403,11 +403,11 @@ def emit_feature_service_lineage(
403403
namespace=namespace,
404404
)
405405

406-
# Emit a RunEvent with OTHER state (metadata registration, not execution)
406+
# Emit a RunEvent with COMPLETE state to create lineage connection
407407
return self._client.emit_run_event(
408408
job_name=job.name,
409409
run_id=str(uuid.uuid4()),
410-
event_type=RunState.OTHER,
410+
event_type=RunState.COMPLETE,
411411
inputs=inputs,
412412
outputs=outputs,
413413
job_facets=job.facets,
@@ -851,7 +851,7 @@ def emit_apply(
851851
result1 = self._client.emit_run_event(
852852
job_name=f"feast_feature_views_{project}",
853853
run_id=str(uuid.uuid4()),
854-
event_type=RunState.OTHER,
854+
event_type=RunState.COMPLETE,
855855
inputs=fv_inputs,
856856
outputs=fv_outputs,
857857
job_facets=job_facets,
@@ -967,7 +967,7 @@ def emit_apply(
967967
result = self._client.emit_run_event(
968968
job_name=f"feature_service_{fs.name}", # Prefix to avoid conflict with dataset
969969
run_id=str(uuid.uuid4()),
970-
event_type=RunState.OTHER,
970+
event_type=RunState.COMPLETE,
971971
inputs=fs_inputs,
972972
outputs=[fs_output],
973973
job_facets=job_facets,

sdk/python/tests/doctest/test_all.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,11 @@ def test_docstrings():
7979
full_name = package.__name__ + "." + name
8080
try:
8181
# https://github.com/feast-dev/feast/issues/5088
82-
if "ikv" not in full_name and "milvus" not in full_name:
82+
if (
83+
"ikv" not in full_name
84+
and "milvus" not in full_name
85+
and "openlineage" not in full_name
86+
):
8387
temp_module = importlib.import_module(full_name)
8488
if is_pkg:
8589
next_packages.append(temp_module)

0 commit comments

Comments
 (0)