@@ -75,7 +75,14 @@ def __init__( # noqa: C901
75
75
name : Optional [str ] = None ,
76
76
features : Optional [List [Feature ]] = None ,
77
77
sources : Optional [
78
- List [Union [BatchFeatureView , StreamFeatureView , RequestSource , FeatureViewProjection ]]
78
+ List [
79
+ Union [
80
+ BatchFeatureView ,
81
+ StreamFeatureView ,
82
+ RequestSource ,
83
+ FeatureViewProjection ,
84
+ ]
85
+ ]
79
86
] = None ,
80
87
udf : Optional [MethodType ] = None ,
81
88
inputs : Optional [
@@ -139,7 +146,9 @@ def __init__( # noqa: C901
139
146
for _ , source in inputs .items ():
140
147
if isinstance (source , FeatureView ):
141
148
_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
+ ):
143
152
_sources .append (source )
144
153
else :
145
154
raise ValueError (
@@ -182,7 +191,9 @@ def __init__( # noqa: C901
182
191
for _ , source in _inputs .items ():
183
192
if isinstance (source , FeatureView ):
184
193
_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
+ ):
186
197
_sources .append (source )
187
198
else :
188
199
raise ValueError (
@@ -493,7 +504,14 @@ def on_demand_feature_view(
493
504
* args ,
494
505
features : Optional [List [Feature ]] = None ,
495
506
sources : Optional [
496
- List [Union [BatchFeatureView , StreamFeatureView , RequestSource , FeatureViewProjection ]]
507
+ List [
508
+ Union [
509
+ BatchFeatureView ,
510
+ StreamFeatureView ,
511
+ RequestSource ,
512
+ FeatureViewProjection ,
513
+ ]
514
+ ]
497
515
] = None ,
498
516
inputs : Optional [Dict [str , Union [FeatureView , RequestSource ]]] = None ,
499
517
schema : Optional [List [Field ]] = None ,
@@ -549,7 +567,9 @@ def on_demand_feature_view(
549
567
for _ , source in inputs .items ():
550
568
if isinstance (source , FeatureView ):
551
569
_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
+ ):
553
573
_sources .append (source )
554
574
else :
555
575
raise ValueError (
@@ -589,7 +609,9 @@ def on_demand_feature_view(
589
609
for _ , source in _inputs .items ():
590
610
if isinstance (source , FeatureView ):
591
611
_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
+ ):
593
615
_sources .append (source )
594
616
else :
595
617
raise ValueError (
0 commit comments