fix! join_keys typing.

Signed-off-by: williamfoschiera <[email protected]>
feast-dev · Jun 5, 2023 · 8a7bb22 · 8a7bb22
1 parent 546019a
commit 8a7bb22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/inference.py b/sdk/python/feast/inference.py
@@ -1,5 +1,5 @@
 import re
-from typing import List, Set, Union
+from typing import List, Optional, Set, Union
 
 from feast.data_source import DataSource, PushSource, RequestSource
 from feast.entity import Entity
@@ -187,7 +187,7 @@ def update_feature_views_with_inferred_features_and_entities(
 
 def _infer_features_and_entities(
     fv: FeatureView,
-    join_keys: Set[str],
+    join_keys: Set[Optional[str]],
     run_inference_for_features,
     config,
 ) -> None: