From e0fc65ec8a90fbf4a9578aa075e2119d456fd33c Mon Sep 17 00:00:00 2001
From: Michael Duong <michaelduong22@gmail.com>
Date: Mon, 3 Nov 2025 17:47:20 +0000
Subject: [PATCH] add conditional imports

---
 modules/ml-pipeline/src/pipeline/core/MLModels.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/modules/ml-pipeline/src/pipeline/core/MLModels.py b/modules/ml-pipeline/src/pipeline/core/MLModels.py
index dabe154..2444314 100644
--- a/modules/ml-pipeline/src/pipeline/core/MLModels.py
+++ b/modules/ml-pipeline/src/pipeline/core/MLModels.py
@@ -11,9 +11,6 @@ import joblib
 import pandas as pd
 from pathlib import Path
 from typing import Union, List
-from sklearn import linear_model
-from sklearn.svm import SVR
-from autogluon.tabular import TabularDataset, TabularPredictor
 from core.interface.InterfaceModels import MLModel
 from core.Logger import logger
 
@@ -69,6 +66,8 @@ class SKLearnLinearRegression:
         """
         Method to train a model
         """
+        from sklearn import linear_model
+
         self.model = linear_model.LinearRegression()
 
         x_train = data.iloc[:, data.columns != target]
@@ -117,6 +116,7 @@ class SKLearnSVMRegression:
         """
         Method to train a model
         """
+        from sklearn.svm import SVR
 
         validate_dict_keys(
             list(model_hyperparameters.keys()),
@@ -161,6 +161,8 @@ class AutogluonAutoML:
         """
         Method to load a model
         """
+        from autogluon.tabular import TabularPredictor
+
         filepath = str(path)
         self.model = TabularPredictor.load(path=filepath)
 
@@ -186,6 +188,8 @@ class AutogluonAutoML:
         """
         Method to train a model
         """
+        from autogluon.tabular import TabularDataset, TabularPredictor
+
         # Force Parallel Model fitting
         os.environ["AG_FORCE_PARALLEL"] = "True"