From d3e0f90399d9985ebb8243447870cfe04e0ac739 Mon Sep 17 00:00:00 2001
From: Michael Skralivetsky <Michael_Skralivetsky@mckinsey.com>
Date: Mon, 26 Aug 2024 21:08:31 +0300
Subject: [PATCH] test-mlflow-tracking.ipynb: improve pip command

---
 test-mlflow-tracking/test-mlflow-tracking.ipynb | 15 +--------------
 1 file changed, 1 insertion(+), 14 deletions(-)

diff --git a/test-mlflow-tracking/test-mlflow-tracking.ipynb b/test-mlflow-tracking/test-mlflow-tracking.ipynb
index a5ef639..5bf7e35 100644
--- a/test-mlflow-tracking/test-mlflow-tracking.ipynb
+++ b/test-mlflow-tracking/test-mlflow-tracking.ipynb
@@ -10,20 +10,7 @@
    },
    "outputs": [],
    "source": [
-    "! pip install mlflow"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "c5d35479-7453-4ec9-84ee-686d7f779561",
-   "metadata": {
-    "scrolled": true,
-    "tags": []
-   },
-   "outputs": [],
-   "source": [
-    "! pip install xgboost"
+    "!pip install mlflow xgboost mlrun[complete]"
    ]
   },
   {