small fix

Author: cristian-martinez-olx

mlflow-project/train_keras.py

@@ -33,7 +33,7 @@
 @click.option("--hidden-units", default=20, type=int)
 def train_keras(ratings_data, als_model_uri, hidden_units):
     np.random.seed(0)
-    tf.set_random_seed(42)  # For reproducibility
+    tf.random.set_seed(42)  # For reproducibility
 
     spark = pyspark.sql.SparkSession.builder.getOrCreate()
     als_model = mlflow.spark.load_model(als_model_uri).stages[0]

notebooks/3-mlflow-models-and-model-registry.ipynb

@@ -53,16 +53,6 @@
     ")[0]"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "run_id, metric_value = best_run('r2', experiment_id=1)\n",
-    "print(f\"best run: '{run_id}', r2-score: {metric_value:.5f}\")"
-   ]
-  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -77,7 +67,7 @@
    "outputs": [],
    "source": [
     "result = mlflow.register_model(\n",
-    "    f\"runs:/{run_id}/model\",\n",
+    "    f\"runs:/{run.info.run_id}/model\",\n",
     "    \"wine-quality-predictor\"\n",
     ")"
    ]