fix: Locate feature_store.yaml from __file__ (feast-dev#4443)

fix: locate feature_store.yaml from __file__ Signed-off-by: Yang, Bo <bo.yang@protonbase.io>
lokeshrangineni · Aug 25, 2024 · 20290ce · 20290ce
1 parent 19cf222
commit 20290ce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sdk/python/feast/templates/postgres/feature_repo/test_workflow.py b/sdk/python/feast/templates/postgres/feature_repo/test_workflow.py
@@ -1,3 +1,4 @@
+import os.path
 import subprocess
 from datetime import datetime
 
@@ -8,7 +9,7 @@
 
 
 def run_demo():
-    store = FeatureStore(repo_path=".")
+    store = FeatureStore(repo_path=os.path.dirname(__file__))
     print("\n--- Run feast apply to setup feature store on Postgres ---")
     subprocess.run(["feast", "apply"])