From 50f94bebd863cbcedf0e6bfb004d4fe7f5c9340d Mon Sep 17 00:00:00 2001
From: ruplisso <44896278+ruplisso@users.noreply.github.com>
Date: Mon, 7 Jun 2021 18:45:37 +0200
Subject: [PATCH] Update code_test.py

---
 tests/unit/code_test.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/unit/code_test.py b/tests/unit/code_test.py
index 6b6d60ac..3ae1c2dd 100644
--- a/tests/unit/code_test.py
+++ b/tests/unit/code_test.py
@@ -1,4 +1,5 @@
 import numpy as np
+import lightgbm
 from azureml.core.run import Run
 from unittest.mock import Mock
 from diabetes_regression.training.train import train_model