Take 59

ExpediaGroup · Feb 14, 2024 · 32f53c3 · 32f53c3
1 parent d0d86bf
commit 32f53c3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sdk/python/tests/unit/infra/test_streaming_ingestion.py b/sdk/python/tests/unit/infra/test_streaming_ingestion.py
@@ -30,6 +30,10 @@
 from tests.integration.feature_repos.universal.online_store.redis import (
     RedisOnlineStoreCreator,
 )
+from feast.data_source import KafkaSource
+from feast.data_format import AvroFormat, ConfluentAvroFormat
+from feast import FileSource
+from feast.stream_feature_view import StreamFeatureView
 
 
 def test_streaming_ingestion():
@@ -51,7 +55,7 @@ def test_streaming_ingestion():
         name="kafka",
         timestamp_field="event_timestamp",
         kafka_bootstrap_servers="",
-        message_format=AvroFormat(""),
+        message_format=ConfluentAvroFormat(""),
         topic="topic",
         batch_source=FileSource(path="some path"),
     )