Chronos: Fix readthedocs.io build error (#4720)

* add pickle5 version * fix doc issue * fix import bigdl.nano in doc * renamed basepytorchmodel to baseautomodel * renamed again
intel-analytics · May 30, 2022 · ebba044 · ebba044
1 parent 367702e
commit ebba044
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/docs/readthedocs/requirements-doc.txt b/docs/readthedocs/requirements-doc.txt
@@ -20,11 +20,13 @@ commonmark==0.8.1
 recommonmark==0.5.0
 readthedocs-sphinx-ext<2.2
 sphinx_rtd_theme==0.5.2
-scikit-learn==0.22.2.post1
+scikit-learn==1.0.2
 pystan==2.19.1.1
 prophet
 pmdarima
 sphinx_markdown_tables
 numpy==1.21.2
 xgboost
-torchmetrics
+torchmetrics
+pickle5==0.0.12
+protobuf~=3.19.0
diff --git a/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst b/docs/readthedocs/source/doc/PythonAPI/Chronos/automodels.rst
@@ -5,7 +5,7 @@ AutoTCN
 -------------------------------------------
 
 AutoTCN is a TCN forecasting model with Auto tuning.
-Other API follows its base class(BasePytorchAutomodel).
+Other API follows its base class(BaseAutomodel).
 
 .. automodule:: bigdl.chronos.autots.model.auto_tcn
     :members:
@@ -17,7 +17,7 @@ AutoLSTM
 ----------------------------------------
 
 AutoLSTM is an LSTM forecasting model with Auto tuning.
-Other API follows its base class(BasePytorchAutomodel).
+Other API follows its base class(BaseAutomodel).
 
 .. automodule:: bigdl.chronos.autots.model.auto_lstm
     :members:
@@ -28,7 +28,7 @@ AutoSeq2Seq
 ----------------------------------------
 
 AutoSeq2Seq is an Seq2Seq forecasting model with Auto tuning.
-Other API follows its base class(BasePytorchAutomodel).
+Other API follows its base class(BaseAutomodel).
 
 .. automodule:: bigdl.chronos.autots.model.auto_seq2seq
     :members:
@@ -55,10 +55,10 @@ AutoProphet is a Prophet forecasting model with Auto tuning.
     :undoc-members:
     :show-inheritance:
 
-BasePytorchAutomodel
+BaseAutomodel
 ------------------------------------------------------------
 AutoLSTM, AutoSeq2Seq and AutoTCN all follow the same API as stated below.
 
-.. autoclass:: bigdl.chronos.autots.model.base_automodel.BasePytorchAutomodel
+.. autoclass:: bigdl.chronos.autots.model.base_automodel.BaseAutomodel
     :members:
     :show-inheritance:
diff --git a/python/chronos/src/bigdl/chronos/autots/tspipeline.py b/python/chronos/src/bigdl/chronos/autots/tspipeline.py
@@ -21,7 +21,6 @@
 
 from bigdl.chronos.data import TSDataset
 from bigdl.chronos.metric.forecast_metrics import Evaluator
-from bigdl.chronos.pytorch.utils import _pytorch_fashion_inference
 
 DEFAULT_MODEL_INIT_DIR = "model_init.ckpt"
 DEFAULT_BEST_MODEL_DIR = "best_model.ckpt"