fix secrets API for 4.x

featurestorebook · Dec 12, 2024 · 0d92a90 · 0d92a90
1 parent d2a8292
commit 0d92a90
Show file tree

Hide file tree

Showing 14 changed files with 53 additions and 4,157 deletions.
diff --git a/notebooks/ch03/1_air_quality_feature_backfill.ipynb b/notebooks/ch03/1_air_quality_feature_backfill.ipynb
@@ -326,7 +326,7 @@
     }
    ],
    "source": [
-    "secrets = util.secrets_api(project.name)\n",
+    "secrets = hopsworks.get_secrets_api()\n",
     "try:\n",
     "    secrets.create_secret(\"AQI_API_KEY\", AQI_API_KEY)\n",
     "except hopsworks.RestAPIError:\n",

diff --git a/notebooks/ch03/2_air_quality_feature_pipeline.ipynb b/notebooks/ch03/2_air_quality_feature_pipeline.ipynb
@@ -96,7 +96,7 @@
     "\n",
     "project = hopsworks.login()\n",
     "fs = project.get_feature_store() \n",
-    "secrets = util.secrets_api(project.name)\n",
+    "secrets = hopsworks.get_secrets_api()\n",
     "\n",
     "# This line will fail if you have not registered the AQI_API_KEY as a secret in Hopsworks\n",
     "AQI_API_KEY = secrets.get_secret(\"AQI_API_KEY\").value\n",

diff --git a/notebooks/ch03/3_air_quality_training_pipeline.ipynb b/notebooks/ch03/3_air_quality_training_pipeline.ipynb
diff --git a/notebooks/ch03/4_air_quality_batch_inference.ipynb b/notebooks/ch03/4_air_quality_batch_inference.ipynb
@@ -91,7 +91,7 @@
     "project = hopsworks.login()\n",
     "fs = project.get_feature_store() \n",
     "\n",
-    "secrets = util.secrets_api(project.name)\n",
+    "secrets = hopsworks.get_secrets_api()\n",
     "location_str = secrets.get_secret(\"SENSOR_LOCATION_JSON\").value\n",
     "location = json.loads(location_str)\n",
     "country=location['country']\n",

diff --git a/notebooks/ch03/functions/util.py b/notebooks/ch03/functions/util.py
@@ -280,14 +280,6 @@ def purge_project(proj):
     delete_models(mr, "air_quality_xgboost_model")
     delete_secrets(proj, "SENSOR_LOCATION_JSON")
 
-
-def secrets_api(proj):
-    host = "c.app.hopsworks.ai"
-    api_key = os.environ.get('HOPSWORKS_API_KEY')
-    conn = hopsworks.connection(host=host, project=proj, api_key_value=api_key)
-    return conn.get_secrets_api()
-
-
 def check_file_path(file_path):
     my_file = Path(file_path)
     if my_file.is_file() == False:

diff --git a/notebooks/ch06/cc-fraud/feature-pipeline/bytewax_profile_activity_5min.py b/notebooks/ch06/cc-fraud/feature-pipeline/bytewax_profile_activity_5min.py
diff --git a/notebooks/ch06/cc-fraud/feature-pipeline/features/transactions.py b/notebooks/ch06/cc-fraud/feature-pipeline/features/transactions.py