Skip to content

Commit

Permalink
Yogi fs bugfix (#2788)
Browse files Browse the repository at this point in the history
* Cleaned up some comments, edited cell metadata, and commented out some code using better materialization.

* Fixed code formatting errors.

* Updated code snippet for attaching Redis instance to feature store as online store.

* Fixed bug BUG 2783996.

* Fixed bug BUG 2783996.

* Removed a redundant code cell.
  • Loading branch information
ynpandey authored Nov 2, 2023
1 parent 7eaf9b2 commit 3fc72ac
Showing 1 changed file with 2 additions and 48 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -390,52 +390,6 @@
")"
]
},
{
"attachments": {},
"cell_type": "markdown",
"metadata": {},
"source": [
"### Retrieve the user-assigned managed identity (UAI) used for feature store for materialization\n",
"This code cell retrieves the principal ID, client ID, and ARM ID property values for the UAI that will be used by the feature store for data materialization."
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {
"gather": {
"logged": 1696554548087
},
"jupyter": {
"outputs_hidden": false,
"source_hidden": false
},
"name": "retrieve-uai",
"nteract": {
"transient": {
"deleting": false
}
}
},
"outputs": [],
"source": [
"from azure.mgmt.msi import ManagedServiceIdentityClient\n",
"\n",
"uai_arm_id = fs_client.feature_stores.get(\n",
" featurestore_name\n",
").materialization_identity.resource_id\n",
"uai_principal_id = fs_client.feature_stores.get(\n",
" featurestore_name\n",
").materialization_identity.principal_id\n",
"uai_client_id = fs_client.feature_stores.get(\n",
" featurestore_name\n",
").materialization_identity.client_id\n",
"\n",
"print(\"uai_principal_id:\" + uai_principal_id)\n",
"print(\"uai_client_id:\" + uai_client_id)\n",
"print(\"uai_arm_id:\" + uai_arm_id)"
]
},
{
"attachments": {},
"cell_type": "markdown",
Expand Down Expand Up @@ -578,7 +532,7 @@
"from datetime import datetime, timedelta\n",
"\n",
"# Trigger backfill on the \"accounts\" feature set.\n",
"# Backfill from 01/01/2023 to all the way to 3 hours ago.\n",
"# Backfill from 01/01/2020 to all the way to 3 hours ago.\n",
"\n",
"st = datetime(2020, 1, 1, 0, 0, 0, 0)\n",
"ed = datetime.now() - timedelta(hours=3)\n",
Expand Down Expand Up @@ -695,7 +649,7 @@
"outputs": [],
"source": [
"# Trigger backfill on the \"transactions\" feature set to fill in the online/offline store.\n",
"# Backfill from 01/01/2023 to all the way to 3 hours ago.\n",
"# Backfill from 01/01/2020 to all the way to 3 hours ago.\n",
"\n",
"from datetime import datetime, timedelta\n",
"\n",
Expand Down

0 comments on commit 3fc72ac

Please sign in to comment.