From e2c82249b37e78bfe8c0cb2f34050db8c06021db Mon Sep 17 00:00:00 2001 From: Rayhao Date: Thu, 26 Sep 2024 21:13:48 -0700 Subject: [PATCH] import issue --- expo/experimenter/autogluon.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/expo/experimenter/autogluon.py b/expo/experimenter/autogluon.py index e5e3045f1..dabf0c138 100644 --- a/expo/experimenter/autogluon.py +++ b/expo/experimenter/autogluon.py @@ -1,6 +1,5 @@ from datetime import datetime from expo.experimenter.custom import CustomExperimenter -from autogluon.tabular import TabularDataset, TabularPredictor import os import pandas as pd @@ -11,6 +10,7 @@ class AGRunner: self.datasets = self.state["datasets_dir"] def run(self): + from autogluon.tabular import TabularDataset, TabularPredictor train_path = self.datasets["train"] dev_path = self.datasets["dev"] dev_wo_target_path = self.datasets["dev_wo_target"]