From 67c41b4168287de18406d712697b4905a2d5f985 Mon Sep 17 00:00:00 2001 From: Achal Shah Date: Mon, 11 Apr 2022 11:55:19 -0700 Subject: [PATCH] fix tests Signed-off-by: Achal Shah --- sdk/python/feast/data_source.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdk/python/feast/data_source.py b/sdk/python/feast/data_source.py index fd981f1ea3..46df1088db 100644 --- a/sdk/python/feast/data_source.py +++ b/sdk/python/feast/data_source.py @@ -704,7 +704,7 @@ def from_proto(data_source: DataSourceProto): for key, val in schema_pb.items(): schema[key] = ValueType(val) - assert data_source.push_options.HasField("batch_source") + assert data_source.HasField("batch_source") batch_source = DataSource.from_proto(data_source.batch_source) return PushSource(