From a39acd33810eea9be681982c7a5fb6ff1d055841 Mon Sep 17 00:00:00 2001
From: GowthamVidiyala <gowtham.vidiyala@gmail.com>
Date: Mon, 8 Mar 2021 17:40:11 +0530
Subject: [PATCH] Update train.py

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 26765faf..80af3c7c 100644
--- a/train.py
+++ b/train.py
@@ -25,6 +25,7 @@ def eval_metrics(actual, pred):
 
 
 if __name__ == "__main__":
+    mlflow.set_tracking_uri("sqlite:///mlruns.db")
     warnings.filterwarnings("ignore")
     np.random.seed(40)