Update Database name for Feature Engineering notebooks (#171)

databricks · Sep 10, 2024 · 267b1e5 · 267b1e5
1 parent dc1dea6
commit 267b1e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...phanumeric_underscore` .}}/feature_engineering/notebooks/GenerateAndWriteFeatures.py.tmpl b/...phanumeric_underscore` .}}/feature_engineering/notebooks/GenerateAndWriteFeatures.py.tmpl
@@ -82,8 +82,8 @@ pk_columns = dbutils.widgets.get("primary_keys")
 assert input_table_path != "", "input_table_path notebook parameter must be specified"
 assert output_table_name != "", "output_table_name notebook parameter must be specified"
 
-# Extract database name. Needs to be updated for Unity Catalog.
-output_database = output_table_name.split(".")[0]
+# Extract database name. Needs to be updated for Unity Catalog to the Schema name.
+output_database = output_table_name.split(".")[1]
 
 # COMMAND ----------