Skip to content

Commit

Permalink
switch to logging with logger
Browse files Browse the repository at this point in the history
  • Loading branch information
emgeee committed Dec 9, 2024
1 parent 72e4525 commit 44c4908
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion py-denormalized/python/denormalized/feast_data_stream.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import inspect
import logging
from typing import Any, TypeVar, Union, cast, get_type_hints

import pyarrow as pa
Expand All @@ -10,6 +11,8 @@

from .data_stream import DataStream

logger = logging.getLogger(__name__)

T = TypeVar("T")


Expand Down Expand Up @@ -113,6 +116,6 @@ def _sink_to_feast(rb: pa.RecordBatch):
try:
feature_store.push(source_name, df, to=PushMode.ONLINE)
except Exception as e:
print(e)
logger.error(f"Failed to push to Feast feature store: {e}", exc_info=True)

self.ds.sink_python(_sink_to_feast)

0 comments on commit 44c4908

Please sign in to comment.