Skip to content

Commit 99e0cf5

Browse files
committed
lilnt
Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
1 parent 4b9dd34 commit 99e0cf5

File tree

1 file changed

+28
-6
lines changed

1 file changed

+28
-6
lines changed

sdk/python/feast/on_demand_feature_view.py

+28-6
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,14 @@ def __init__( # noqa: C901
7575
name: Optional[str] = None,
7676
features: Optional[List[Feature]] = None,
7777
sources: Optional[
78-
List[Union[BatchFeatureView, StreamFeatureView, RequestSource, FeatureViewProjection]]
78+
List[
79+
Union[
80+
BatchFeatureView,
81+
StreamFeatureView,
82+
RequestSource,
83+
FeatureViewProjection,
84+
]
85+
]
7986
] = None,
8087
udf: Optional[MethodType] = None,
8188
inputs: Optional[
@@ -139,7 +146,9 @@ def __init__( # noqa: C901
139146
for _, source in inputs.items():
140147
if isinstance(source, FeatureView):
141148
_sources.append(feature_view_to_batch_feature_view(source))
142-
elif isinstance(source, RequestSource) or isinstance(source, FeatureViewProjection):
149+
elif isinstance(source, RequestSource) or isinstance(
150+
source, FeatureViewProjection
151+
):
143152
_sources.append(source)
144153
else:
145154
raise ValueError(
@@ -182,7 +191,9 @@ def __init__( # noqa: C901
182191
for _, source in _inputs.items():
183192
if isinstance(source, FeatureView):
184193
_sources.append(feature_view_to_batch_feature_view(source))
185-
elif isinstance(source, RequestSource) or isinstance(source, FeatureViewProjection):
194+
elif isinstance(source, RequestSource) or isinstance(
195+
source, FeatureViewProjection
196+
):
186197
_sources.append(source)
187198
else:
188199
raise ValueError(
@@ -493,7 +504,14 @@ def on_demand_feature_view(
493504
*args,
494505
features: Optional[List[Feature]] = None,
495506
sources: Optional[
496-
List[Union[BatchFeatureView, StreamFeatureView, RequestSource, FeatureViewProjection]]
507+
List[
508+
Union[
509+
BatchFeatureView,
510+
StreamFeatureView,
511+
RequestSource,
512+
FeatureViewProjection,
513+
]
514+
]
497515
] = None,
498516
inputs: Optional[Dict[str, Union[FeatureView, RequestSource]]] = None,
499517
schema: Optional[List[Field]] = None,
@@ -549,7 +567,9 @@ def on_demand_feature_view(
549567
for _, source in inputs.items():
550568
if isinstance(source, FeatureView):
551569
_sources.append(feature_view_to_batch_feature_view(source))
552-
elif isinstance(source, RequestSource) or isinstance(source, FeatureViewProjection):
570+
elif isinstance(source, RequestSource) or isinstance(
571+
source, FeatureViewProjection
572+
):
553573
_sources.append(source)
554574
else:
555575
raise ValueError(
@@ -589,7 +609,9 @@ def on_demand_feature_view(
589609
for _, source in _inputs.items():
590610
if isinstance(source, FeatureView):
591611
_sources.append(feature_view_to_batch_feature_view(source))
592-
elif isinstance(source, RequestSource) or isinstance(source, FeatureViewProjection):
612+
elif isinstance(source, RequestSource) or isinstance(
613+
source, FeatureViewProjection
614+
):
593615
_sources.append(source)
594616
else:
595617
raise ValueError(

0 commit comments

Comments
 (0)