upgrade feature_store.yaml

adchia · adchia · commit ce6b0851d6bf · 2022-09-29T10:37:18.000-05:00
Signed-off-by: Danny Chiao &lt;danny@tecton.ai&gt;
diff --git a/.gitignore b/.gitignore
@@ -4,4 +4,5 @@
 **/__pycache__
 terraform.tfstate
 terraform.tfstate.backup
-.terraform*
+.terraform*
+*.iml
diff --git a/module_0/client_aws/feature_store.yaml b/module_0/client_aws/feature_store.yaml
@@ -3,4 +3,5 @@ provider: aws
 registry: s3://feast-workshop-danny/registry.pb # TODO: Replace with your bucket
 online_store: null
 offline_store:
-  type: file
+  type: file
+entity_key_serialization_version: 2
diff --git a/module_0/client_gcp/feature_store.yaml b/module_0/client_gcp/feature_store.yaml
@@ -3,4 +3,5 @@ provider: gcp
 registry: gs://feast-workshop-danny/registry.pb # TODO: Replace with your bucket
 online_store: null
 offline_store:
-  type: file
+  type: file
+entity_key_serialization_version: 2
diff --git a/module_0/feature_repo_aws/feature_store.yaml b/module_0/feature_repo_aws/feature_store.yaml
@@ -3,4 +3,5 @@ provider: aws
 registry: s3://feast-workshop-danny/registry.pb # TODO: Replace with your bucket
 online_store: null
 offline_store:
-  type: file
+  type: file
+entity_key_serialization_version: 2
diff --git a/module_0/feature_repo_gcp/feature_store.yaml b/module_0/feature_repo_gcp/feature_store.yaml
@@ -3,4 +3,5 @@ provider: gcp
 registry: gs://feast-workshop-danny/registry.pb # TODO: Replace with your bucket
 online_store: null
 offline_store:
-  type: file
+  type: file
+entity_key_serialization_version: 2
diff --git a/module_1/feature_repo/entities.py b/module_1/feature_repo/entities.py
@@ -1,11 +1,9 @@
 from feast import (
     Entity,
-    ValueType,
 )
 
 driver = Entity(
     name="driver",
     join_keys=["driver_id"],
-    value_type=ValueType.INT64,
     description="driver id",
 )
diff --git a/module_1/feature_repo/feature_store.yaml b/module_1/feature_repo/feature_store.yaml
@@ -8,3 +8,5 @@ online_store:
   connection_string: localhost:6379
 offline_store:
   type: file
+entity_key_serialization_version: 2
+
diff --git a/module_1/feature_repo/features.py b/module_1/feature_repo/features.py
@@ -4,16 +4,19 @@
     FeatureView,
     Field,
 )
-from feast.types import Float32
+from feast.types import Float32, Int64
 
 from data_sources import *
 from entities import *
 
 driver_daily_features_view = FeatureView(
     name="driver_daily_features",
-    entities=["driver"],
+    entities=[driver],
     ttl=timedelta(seconds=8640000000),
-    schema=[Field(name="daily_miles_driven", dtype=Float32),],
+    schema=[
+        Field(name="driver_id", dtype=Int64),
+        Field(name="daily_miles_driven", dtype=Float32),
+    ],
     online=True,
     source=driver_stats_push_source,
     tags={"production": "True"},
@@ -23,9 +26,10 @@
 driver_hourly_stats_view = FeatureView(
     name="driver_hourly_stats",
     description="Hourly features",
-    entities=["driver"],
+    entities=[driver],
     ttl=timedelta(seconds=8640000000),
     schema=[
+        Field(name="driver_id", dtype=Int64),
         Field(name="conv_rate", dtype=Float32),
         Field(name="acc_rate", dtype=Float32),
         Field(name="miles_driven", dtype=Float32),

Original file line number	Diff line number	Diff line change
`@@ -1,11 +1,9 @@`
`1`	`1`	`from feast import (`
`2`	`2`	`Entity,`
`3`		`- ValueType,`
`4`	`3`	`)`
`5`	`4`
`6`	`5`	`driver = Entity(`
`7`	`6`	`name="driver",`
`8`	`7`	`join_keys=["driver_id"],`
`9`		`- value_type=ValueType.INT64,`
`10`	`8`	`description="driver id",`
`11`	`9`	`)`