fix lint with isort

docs/conf.py

@@ -12,10 +12,12 @@
 
 import os
 import sys
+
 import sphinx_rtd_theme
 
 sys.path.insert(0, os.path.abspath(".."))
 import matplotlib
+
 import orbit
 
 # -- Project information -----------------------------------------------------

orbit/constants/palette.py

@@ -1,6 +1,7 @@
 from enum import Enum
-import seaborn as sns
+
 import matplotlib.colors as clr
+import seaborn as sns
 
 
 class KTRPalette(Enum):

orbit/diagnostics/backtest.py

@@ -1,15 +1,17 @@
-import pandas as pd
-import numpy as np
-import matplotlib.pyplot as plt
-from copy import deepcopy
 import inspect
+from copy import deepcopy
+
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
 
-from .metrics import smape, wmape, mape, mse, mae, rmsse
-from ..exceptions import BacktestException
-from ..constants.constants import TimeSeriesSplitSchemeKeys, BacktestFitKeys
-from ..constants.palette import OrbitPalette as OrbitPal
 from orbit.utils.plot import orbit_style_decorator
 
+from ..constants.constants import BacktestFitKeys, TimeSeriesSplitSchemeKeys
+from ..constants.palette import OrbitPalette as OrbitPal
+from ..exceptions import BacktestException
+from .metrics import mae, mape, mse, rmsse, smape, wmape
+
 
 class TimeSeriesSplitter(object):
     """Cross validation splitter for time series data"""

orbit/diagnostics/plot.py

@@ -1,27 +1,26 @@
 # the following lines are added to fix unit test error
 # or else the following line will give the following error
 # TclError: no display name and no $DISPLAY environment variable
-import matplotlib.pyplot as plt
-import seaborn as sns
-import pandas as pd
-import numpy as np
+import logging
 import math
 import os
+
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
 import pkg_resources
+import seaborn as sns
 import statsmodels.api as sm
 from scipy import stats
 
-from ..constants.constants import PredictionKeys
-from orbit.utils.general import is_empty_dataframe, is_ordered_datetime
-from ..constants.constants import BacktestFitKeys
-from ..constants.palette import PredictionPaletteClassic as PredPal
 from orbit.constants import palette
 from orbit.diagnostics.metrics import smape
+from orbit.utils.general import is_empty_dataframe, is_ordered_datetime
 from orbit.utils.plot import orbit_style_decorator
-from ..exceptions import PlotException
 
-
-import logging
+from ..constants.constants import BacktestFitKeys, PredictionKeys
+from ..constants.palette import PredictionPaletteClassic as PredPal
+from ..exceptions import PlotException
 
 logger = logging.getLogger("orbit")
 

orbit/eda/eda_plot.py

@@ -1,7 +1,8 @@
+import math
+
+import pandas as pd
 import seaborn as sns
 from matplotlib import pyplot as plt
-import pandas as pd
-import math
 
 from orbit.constants import palette
 from orbit.utils.plot import orbit_style_decorator

orbit/estimators/base_estimator.py

@@ -1,4 +1,5 @@
 from abc import abstractmethod
+
 import numpy as np
 
 # from ..utils.docstring_style import merge_numpy_docs_dedup

orbit/estimators/pyro_estimator.py

@@ -1,15 +1,15 @@
-from abc import abstractmethod
-import numpy as np
 import logging
+from abc import abstractmethod
 
+import numpy as np
 import pyro
 from pyro.infer import SVI, Trace_ELBO
-from pyro.infer.autoguide import AutoLowRankMultivariateNormal, AutoDelta
+from pyro.infer.autoguide import AutoDelta, AutoLowRankMultivariateNormal
 from pyro.optim import ClippedAdam
 
-from .base_estimator import BaseEstimator
 from ..exceptions import EstimatorException
 from ..utils.pyro import get_pyro_model
+from .base_estimator import BaseEstimator
 
 logger = logging.getLogger("orbit")
 

orbit/estimators/stan_estimator.py

@@ -1,15 +1,14 @@
-from abc import abstractmethod
-from copy import copy
 import logging
 import multiprocessing
+from abc import abstractmethod
+from copy import copy
 from sys import platform, version_info
 
-from .base_estimator import BaseEstimator
 from ..exceptions import EstimatorException
-from ..utils.stan import get_compiled_stan_model, suppress_stdout_stderr
 from ..utils.general import update_dict
 from ..utils.logger import get_logger
-
+from ..utils.stan import get_compiled_stan_model, suppress_stdout_stderr
+from .base_estimator import BaseEstimator
 
 logger = get_logger("orbit")
 

orbit/forecaster/__init__.py

@@ -1,4 +1,4 @@
+from .forecaster import *
+from .full_bayes import *
 from .map import *
 from .svi import *
-from .full_bayes import *
-from .forecaster import *

orbit/forecaster/forecaster.py

@@ -1,16 +1,17 @@
-from copy import deepcopy
-import numpy as np
-import pandas as pd
 import warnings
+from copy import deepcopy
 from enum import Enum
+from typing import Any, Dict
 
-from typing import Dict, Any
+import numpy as np
+import pandas as pd
 
-from ..exceptions import ForecasterException, AbstractMethodException, IllegalArgument
-from ..utils.general import is_ordered_datetime, is_even_gap_datetime
-from ..template.model_template import ModelTemplate
+from ..constants.constants import PredictionMetaKeys, TrainingMetaKeys
 from ..estimators.stan_estimator import StanEstimatorMCMC
-from ..constants.constants import TrainingMetaKeys, PredictionMetaKeys
+from ..exceptions import (AbstractMethodException, ForecasterException,
+                          IllegalArgument)
+from ..template.model_template import ModelTemplate
+from ..utils.general import is_even_gap_datetime, is_ordered_datetime
 
 COMMON_MODEL_CALLABLES = [
     "get_data_input_mapper",

orbit/forecaster/full_bayes.py

@@ -1,10 +1,11 @@
+from functools import partial
+
 import numpy as np
 import pandas as pd
-from functools import partial
 
-from ..constants.constants import PredictMethod, PredictionKeys
+from ..constants.constants import PredictionKeys, PredictMethod
 from ..exceptions import ForecasterException
-from ..utils.predictions import prepend_date_column, compute_percentiles
+from ..utils.predictions import compute_percentiles, prepend_date_column
 from .forecaster import Forecaster
 
 

orbit/forecaster/map.py

@@ -1,10 +1,11 @@
+from functools import partial
+
 import numpy as np
 import pandas as pd
-from functools import partial
 
-from ..constants.constants import PredictMethod, PredictionKeys
+from ..constants.constants import PredictionKeys, PredictMethod
 from ..exceptions import ForecasterException
-from ..utils.predictions import prepend_date_column, compute_percentiles
+from ..utils.predictions import compute_percentiles, prepend_date_column
 from .forecaster import Forecaster
 
 

orbit/forecaster/svi.py

@@ -1,10 +1,11 @@
+from functools import partial
+
 import numpy as np
 import pandas as pd
-from functools import partial
 
-from ..constants.constants import PredictMethod, PredictionKeys
+from ..constants.constants import PredictionKeys, PredictMethod
 from ..exceptions import ForecasterException
-from ..utils.predictions import prepend_date_column, compute_percentiles
+from ..utils.predictions import compute_percentiles, prepend_date_column
 from .forecaster import Forecaster
 
 

orbit/models/__init__.py

@@ -1,5 +1,5 @@
-from .ets import ETS
-from .lgt import LGT
 from .dlt import DLT
-from .ktrlite import KTRLite
+from .ets import ETS
 from .ktr import KTR
+from .ktrlite import KTRLite
+from .lgt import LGT

orbit/models/dlt.py

@@ -1,8 +1,8 @@
 # from inspect import signature
-from ..template.dlt import DLTModel
-from ..forecaster import MAPForecaster, FullBayesianForecaster
-from ..exceptions import IllegalArgument
 from ..constants.constants import EstimatorsKeys
+from ..exceptions import IllegalArgument
+from ..forecaster import FullBayesianForecaster, MAPForecaster
+from ..template.dlt import DLTModel
 
 
 def DLT(

orbit/models/ets.py

@@ -1,8 +1,8 @@
 # from inspect import signature
-from ..template.ets import ETSModel
-from ..forecaster import MAPForecaster, FullBayesianForecaster
-from ..exceptions import IllegalArgument
 from ..constants.constants import EstimatorsKeys
+from ..exceptions import IllegalArgument
+from ..forecaster import FullBayesianForecaster, MAPForecaster
+from ..template.ets import ETSModel
 
 
 def ETS(

orbit/models/ktr.py

@@ -1,8 +1,8 @@
-from ..template.ktr import KTRModel
-from ..forecaster import SVIForecaster
-from ..exceptions import IllegalArgument
-from ..estimators.pyro_estimator import PyroEstimatorSVI
 from ..constants.constants import EstimatorsKeys
+from ..estimators.pyro_estimator import PyroEstimatorSVI
+from ..exceptions import IllegalArgument
+from ..forecaster import SVIForecaster
+from ..template.ktr import KTRModel
 
 
 def KTR(

orbit/models/ktrlite.py

@@ -1,8 +1,8 @@
-from ..template.ktrlite import KTRLiteModel
-from ..forecaster import MAPForecaster
-from ..exceptions import IllegalArgument
-from ..estimators.stan_estimator import StanEstimatorMAP
 from ..constants.constants import EstimatorsKeys
+from ..estimators.stan_estimator import StanEstimatorMAP
+from ..exceptions import IllegalArgument
+from ..forecaster import MAPForecaster
+from ..template.ktrlite import KTRLiteModel
 
 
 def KTRLite(

orbit/models/lgt.py

@@ -1,8 +1,8 @@
 # from inspect import signature
-from ..template.lgt import LGTModel
-from ..forecaster import MAPForecaster, FullBayesianForecaster, SVIForecaster
-from ..exceptions import IllegalArgument
 from ..constants.constants import EstimatorsKeys
+from ..exceptions import IllegalArgument
+from ..forecaster import FullBayesianForecaster, MAPForecaster, SVIForecaster
+from ..template.lgt import LGTModel
 
 
 def LGT(

orbit/pyro/ktr.py

@@ -1,9 +1,7 @@
 import numpy as np
-import torch
-
 import pyro
 import pyro.distributions as dist
-
+import torch
 
 # FIXME: this is sort of dangerous; consider better implementation later
 torch.set_default_tensor_type("torch.DoubleTensor")

orbit/pyro/lgt.py

@@ -1,8 +1,7 @@
 import numpy as np
-import torch
-
 import pyro
 import pyro.distributions as dist
+import torch
 
 torch.set_default_tensor_type("torch.DoubleTensor")
 

orbit/template/dlt.py

@@ -1,23 +1,19 @@
+from copy import deepcopy
+from enum import Enum
+
 import numpy as np
 import pandas as pd
-from scipy.stats import nct
-from copy import deepcopy
 import torch
-from enum import Enum
+from scipy.stats import nct
 
-from ..constants.constants import (
-    DEFAULT_REGRESSOR_SIGN,
-    DEFAULT_REGRESSOR_BETA,
-    DEFAULT_REGRESSOR_SIGMA,
-    COEFFICIENT_DF_COLS,
-    PredictMethod,
-    PredictionKeys,
-    TrainingMetaKeys,
-    PredictionMetaKeys,
-)
+from ..constants.constants import (COEFFICIENT_DF_COLS, DEFAULT_REGRESSOR_BETA,
+                                   DEFAULT_REGRESSOR_SIGMA,
+                                   DEFAULT_REGRESSOR_SIGN, PredictionKeys,
+                                   PredictionMetaKeys, PredictMethod,
+                                   TrainingMetaKeys)
+from ..estimators.stan_estimator import StanEstimatorMAP, StanEstimatorMCMC
 from ..exceptions import IllegalArgument, ModelException, PredictionException
 from .ets import ETSModel
-from ..estimators.stan_estimator import StanEstimatorMCMC, StanEstimatorMAP
 
 
 class DataInputMapper(Enum):

orbit/template/ets.py

@@ -1,13 +1,15 @@
-import numpy as np
 from copy import deepcopy
-import torch
 from enum import Enum
 
-from ..constants.constants import PredictionKeys, TrainingMetaKeys, PredictionMetaKeys
-from ..exceptions import IllegalArgument, DataInputException
-from .model_template import ModelTemplate
-from ..estimators.stan_estimator import StanEstimatorMCMC, StanEstimatorMAP
+import numpy as np
+import torch
+
+from ..constants.constants import (PredictionKeys, PredictionMetaKeys,
+                                   TrainingMetaKeys)
+from ..estimators.stan_estimator import StanEstimatorMAP, StanEstimatorMCMC
+from ..exceptions import DataInputException, IllegalArgument
 from ..utils.features import moving_average
+from .model_template import ModelTemplate
 
 
 # constants for attributes, params and I/Os