@@ -96,29 +96,27 @@ def python_native_test_view(input_dict: dict[str, Any]) -> dict[str, Any]:
9696
9797 python_native_test_view .infer_features ()
9898
99-
100- def test_on_demand_features_invalid_type_inference ():
101- # Create Feature Views
102- date_request = RequestSource (
103- name = "date_request" ,
104- schema = [Field (name = "some_date" , dtype = UnixTimestamp )],
105- )
106-
10799 @on_demand_feature_view (
108100 sources = [date_request ],
109101 schema = [
110102 Field (name = "output" , dtype = UnixTimestamp ),
111103 Field (name = "object_output" , dtype = String ),
112104 ],
113105 )
114- def invalid_test_view (features_df : pd .DataFrame ) -> pd .DataFrame :
106+ def object_string_test_view (features_df : pd .DataFrame ) -> pd .DataFrame :
115107 data = pd .DataFrame ()
116108 data ["output" ] = features_df ["some_date" ]
117109 data ["object_output" ] = features_df ["some_date" ].astype (str )
118110 return data
119111
120- with pytest .raises (ValueError , match = "Value with native type object" ):
121- invalid_test_view .infer_features ()
112+ object_string_test_view .infer_features ()
113+
114+
115+ def test_on_demand_features_invalid_type_inference ():
116+ date_request = RequestSource (
117+ name = "date_request" ,
118+ schema = [Field (name = "some_date" , dtype = UnixTimestamp )],
119+ )
122120
123121 @on_demand_feature_view (
124122 schema = [
@@ -184,14 +182,13 @@ def test_view(features_df: pd.DataFrame) -> pd.DataFrame:
184182 Field (name = "object_output" , dtype = String ),
185183 ],
186184 )
187- def invalid_test_view (features_df : pd .DataFrame ) -> pd .DataFrame :
185+ def object_string_view (features_df : pd .DataFrame ) -> pd .DataFrame :
188186 data = pd .DataFrame ()
189187 data ["output" ] = features_df ["some_date" ]
190188 data ["object_output" ] = features_df ["some_date" ].astype (str )
191189 return data
192190
193- with pytest .raises (ValueError , match = "Value with native type object" ):
194- invalid_test_view .infer_features ()
191+ object_string_view .infer_features ()
195192
196193 @on_demand_feature_view (
197194 sources = [date_request ],
0 commit comments