From c1f2b3ccae35344f1723b6a370e69c293c837f05 Mon Sep 17 00:00:00 2001 From: Ying Chen Date: Tue, 24 Sep 2024 22:22:24 -0700 Subject: [PATCH] Remove unused imports --- runtime/tests/automl_runtime/forecast/deepar/model_test.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/runtime/tests/automl_runtime/forecast/deepar/model_test.py b/runtime/tests/automl_runtime/forecast/deepar/model_test.py index d51201aa..f896688a 100644 --- a/runtime/tests/automl_runtime/forecast/deepar/model_test.py +++ b/runtime/tests/automl_runtime/forecast/deepar/model_test.py @@ -18,19 +18,15 @@ import mlflow import pandas as pd -import pytest import torch import torch.nn as nn -from deepdiff.helper import times from gluonts.dataset.field_names import FieldName from gluonts.transform import InstanceSplitter, TestSplitSampler from gluonts.torch.model.predictor import PyTorchPredictor -from mlflow.exceptions import MlflowException from databricks.automl_runtime.forecast.deepar.model import ( DeepARModel, mlflow_deepar_log_model, ) -from scipy.linalg.interpolative import id_to_svd class TestDeepARModel(unittest.TestCase):