From 32f53c30bdb80b8488d0319b8bab874beeafb047 Mon Sep 17 00:00:00 2001
From: Matt Carlin <matcarlin@expediagroup.com>
Date: Wed, 14 Feb 2024 15:01:21 -0600
Subject: [PATCH] Take 59

---
 sdk/python/tests/unit/infra/test_streaming_ingestion.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/sdk/python/tests/unit/infra/test_streaming_ingestion.py b/sdk/python/tests/unit/infra/test_streaming_ingestion.py
index bdd745c90c..259e16a3f5 100644
--- 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"),
     )