Fix test cases caused by invalid key type (#897)

enya-yx · enya-yx · web-flow · commit 90a6b6d755c3 · 2022-12-02T22:34:41.000-08:00
Co-authored-by: enya-yx &lt;enya@PBHATW10P14S.northamerica.corp.microsoft.com&gt;
diff --git a/feathr_project/test/test_fixture.py b/feathr_project/test/test_fixture.py
@@ -466,7 +466,7 @@ def time_partition_pattern_feature_join_test_setup(config_path: str, data_source
                           postfix_path=postfix_path
                         )
     tpp_key = TypedKey(key_column="f_location_max_fare",
-                    key_column_type=FLOAT)
+                    key_column_type=ValueType.FLOAT)
     tpp_features = [
     Feature(name="key0",
             key=tpp_key,

Original file line number	Diff line number	Diff line change
`@@ -466,7 +466,7 @@ def time_partition_pattern_feature_join_test_setup(config_path: str, data_source`
`466`	`466`	`postfix_path=postfix_path`
`467`	`467`	`)`
`468`	`468`	`tpp_key = TypedKey(key_column="f_location_max_fare",`
`469`		`- key_column_type=FLOAT)`
	`469`	`+ key_column_type=ValueType.FLOAT)`
`470`	`470`	`tpp_features = [`
`471`	`471`	`Feature(name="key0",`
`472`	`472`	`key=tpp_key,`