@@ -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