[Auto] Add deprecation warning. Remove autogluon dependency from esti…

…mators (dmlc#1718) * deprecate * fix * fix lint Co-authored-by: Weisu Yin <[email protected]>
bladesaber · Dec 21, 2021 · 6818c32 · 6818c32
1 parent 3892d7e
commit 6818c32
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 15 deletions.
diff --git a/gluoncv/auto/__init__.py b/gluoncv/auto/__init__.py
@@ -1,2 +1,11 @@
 """GluonCV auto"""
+import logging
 from .estimators import *
+
+logger = logging.getLogger(__name__)
+msg = (
+    "We plan to deprecate auto from gluoncv on release 0.12.0."
+    "Please consider using autogluon.vision instead, which provides the same functionality."
+    "https://auto.gluon.ai/stable/tutorials/image_prediction/index.html"
+)
+logger.warning(msg)
diff --git a/gluoncv/auto/estimators/constants.py b/gluoncv/auto/estimators/constants.py
@@ -0,0 +1,3 @@
+BINARY = 'binary'
+MULTICLASS = 'multiclass'
+REGRESSION = 'regression'
diff --git a/gluoncv/auto/estimators/image_classification/image_classification.py b/gluoncv/auto/estimators/image_classification/image_classification.py
@@ -19,18 +19,12 @@
 from .... import nn
 from .... import loss
 from ..base_estimator import BaseEstimator, set_default
+from ..constants import BINARY, MULTICLASS, REGRESSION
 from .utils import get_data_loader, get_data_rec, smooth, mixup_transform
 from .default import ImageClassificationCfg
 from ...data.dataset import ImageClassificationDataset
 from ..conf import _BEST_CHECKPOINT_FILE
 from ..utils import EarlyStopperOnPlateau
-from ....utils.filesystem import try_import
-problem_type_constants = try_import(package='autogluon.core.constants',
-                                    fromlist=['MULTICLASS', 'BINARY', 'REGRESSION'],
-                                    message='Failed to import problem type constants from autogluon.core.')
-MULTICLASS = problem_type_constants.MULTICLASS
-BINARY = problem_type_constants.BINARY
-REGRESSION = problem_type_constants.REGRESSION
 
 __all__ = ['ImageClassificationEstimator']
 

diff --git a/gluoncv/auto/estimators/torch_image_classification/torch_image_classification.py b/gluoncv/auto/estimators/torch_image_classification/torch_image_classification.py
@@ -27,17 +27,11 @@
 from .default import TorchImageClassificationCfg
 from .utils import resolve_data_config, update_cfg, optimizer_kwargs, \
                    create_scheduler, rmse, create_optimizer_v2a
+from ..constants import BINARY, MULTICLASS, REGRESSION
 from ..utils import EarlyStopperOnPlateau
 from ..utils import _suggest_load_context
 from ..conf import _BEST_CHECKPOINT_FILE
 from ..base_estimator import BaseEstimator, set_default
-from ....utils.filesystem import try_import
-problem_type_constants = try_import(package='autogluon.core.constants',
-                                    fromlist=['MULTICLASS', 'BINARY', 'REGRESSION'],
-                                    message='Failed to import problem type constants from autogluon.core.')
-MULTICLASS = problem_type_constants.MULTICLASS
-BINARY = problem_type_constants.BINARY
-REGRESSION = problem_type_constants.REGRESSION
 
 warnings.filterwarnings('ignore', message='.*Argument interpolation should be of type InterpolationMode instead of int.*')
 logging.getLogger('timm.models').setLevel(logging.WARNING)

diff --git a/setup.py b/setup.py
@@ -73,7 +73,7 @@ def find_version(*file_paths):
 
 extra_requirements = {
     'full': ['tensorboardx', 'decord', 'autogluon.core', 'cython', 'pycocotools'],
-    'auto': ['autogluon.core']
+    'auto': ['autogluon.core==0.3.1']
 }
 
 if with_cython: