upgrade pyre version in fbcode/kats - batch 1

Differential Revision: D55200657

fbshipit-source-id: e290e8634c7abce24361c10dab3193fdf227f4c4

kats/detectors/distribution_distance_model.py

@@ -27,6 +27,7 @@
 
 # Supported metrics for calculating distance
 # details: https://github.com/scipy/scipy/blob/v1.10.1/scipy/spatial/distance.py#L1947
+# pyre-fixme[21]: Could not find module `scipy.spatial.distance`.
 from scipy.spatial.distance import _METRICS_NAMES as SUPPORTED_DISTANCE_METRICS
 
 

kats/detectors/interval_detector.py

@@ -53,6 +53,8 @@
 from kats.detectors.detector import DetectorModel
 from kats.detectors.detector_consts import AnomalyResponse, ConfidenceBand
 from matplotlib import pyplot as plt
+
+# pyre-fixme[21]: Could not find module `numpy.linalg`.
 from numpy.linalg import matrix_power
 from scipy.linalg import toeplitz
 from scipy.stats import beta, binom, mvn, norm

kats/models/bayesian_var.py

@@ -26,6 +26,8 @@
 import numpy as np
 import pandas as pd
 from kats.consts import _log_error, Params, TimeSeriesData
+
+# pyre-fixme[21]: Could not find module `numpy.linalg`.
 from numpy.linalg import inv
 from scipy.linalg import block_diag
 

kats/models/ml_ar.py

@@ -18,6 +18,8 @@
 from kats.utils.parameter_tuning_utils import (
     get_default_lightgbm_parameter_search_space,
 )
+
+# pyre-fixme[21]: Could not find module `numpy.random`.
 from numpy.random import RandomState
 
 try:

kats/tests/detectors/test_bocpd.py

@@ -22,6 +22,8 @@
     TrendChangeParameters,
 )
 from kats.utils.simulator import Simulator
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_cusum_detection.py

@@ -18,6 +18,8 @@
     MultiCUSUMDetector,
     VectorizedCUSUMDetector,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 from scipy.stats import chi2  # @manual
 from sklearn.datasets import make_spd_matrix

kats/tests/detectors/test_cusum_model.py

@@ -23,6 +23,8 @@
     CusumScoreFunction,
     VectorizedCUSUMDetectorModel,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_detector_consts.py

@@ -20,6 +20,8 @@
     PercentageChange,
     SingleSpike,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_distribution_distance_model.py

@@ -20,6 +20,8 @@
     TimeSeriesData,
 )
 from kats.detectors.distribution_distance_model import DistributionDistanceModel
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_interval_detector.py

@@ -26,6 +26,8 @@
     TwoSampleRealValuedIntervalDetectorModel,
     TwoSampleSchema,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 from scipy.stats import norm
 

kats/tests/detectors/test_prophet_detector.py

@@ -23,6 +23,8 @@
     to_seasonality,
 )
 from kats.utils.simulator import Simulator
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 START_DATE_TEST_DATA = "2018-01-01"

kats/tests/detectors/test_robust_stat_detection.py

@@ -17,6 +17,8 @@
     RobustStatChangePoint,
     RobustStatDetector,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 from sklearn.datasets import make_spd_matrix
 

kats/tests/detectors/test_rolling_stats_model.py

@@ -19,6 +19,8 @@
     TimeSeriesData,
 )
 from kats.detectors.rolling_stats_model import RollingStatsModel
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_stat_sig_detector.py

@@ -27,6 +27,8 @@
 )
 from kats.utils.decomposition import SeasonalityHandler
 from kats.utils.simulator import Simulator
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/detectors/test_trend_mk.py

@@ -11,6 +11,8 @@
 import pandas as pd
 from kats.detectors.trend_mk import MKDetector
 from kats.tests.detectors.utils import gen_no_trend_data_ndim, gen_trend_data_ndim
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/metrics/test_metadata.py

@@ -10,6 +10,8 @@
 
 from kats.metrics import metadata
 from kats.metrics.metadata import inspect, MetricMetadata
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/metrics/test_metrics.py

@@ -9,8 +9,12 @@
 from unittest import TestCase
 
 import numpy as np
+
+# pyre-fixme[21]: Could not find module `numpy.testing`.
 import numpy.testing as npt
 from kats.metrics import metrics
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_arima_model.py

@@ -26,6 +26,8 @@
     PEYTON_FCST_30_ARIMA_PARAM_2_MODEL_1,
     PEYTON_FCST_30_ARIMA_PARAM_2_MODEL_2,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 STEPS_1 = 15

kats/tests/models/test_ensemble.py

@@ -27,6 +27,8 @@
 from kats.models.ensemble.kats_ensemble import KatsEnsemble
 from kats.models.ensemble.median_ensemble import MedianEnsembleModel
 from kats.models.ensemble.weighted_avg_ensemble import WeightedAvgEnsemble
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 np.random.seed(123321)

kats/tests/models/test_globalmodel.py

@@ -38,6 +38,8 @@
     S2Cell,
     split,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_holt_winters_model.py

@@ -19,6 +19,8 @@
     AIR_FCST_HW_1,  # first param combination results
     AIR_FCST_HW_2,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 pd_ver = float(re.findall("([0-9]+\\.[0-9]+)\\..*", pd.__version__)[0])

kats/tests/models/test_linear_model.py

@@ -25,6 +25,8 @@
     PEYTON_FCST_LINEAR_NAN,
     PEYTON_INPUT_NAN,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 TEST_DATA: Dict[

kats/tests/models/test_ml_ar_model.py

@@ -12,6 +12,8 @@
 import pandas as pd
 from kats.consts import TimeSeriesData
 from kats.models.ml_ar import MLARModel, MLARParams
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 single_univariate_ts = [

kats/tests/models/test_models_dummy_data.py

@@ -11,6 +11,7 @@
 
 import numpy as np
 import pandas as pd
+# pyre-fixme[21]: Could not find name `Timestamp` in `pandas`.
 from pandas import Timestamp
 
 AIR_FCST_LINEAR_95 = pd.DataFrame(

kats/tests/models/test_neuralprophet_model.py

@@ -22,6 +22,8 @@
     NONSEASONAL_FUTURE_DF,
     NONSEASONAL_INPUT,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_prophet_model.py

@@ -42,6 +42,8 @@
     PEYTON_FCST_30_PROPHET_DAILY_CAP_SM_11,
     PEYTON_FCST_30_PROPHET_DAILY_CAP_SM_12,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_sarima_model.py

@@ -28,6 +28,8 @@
     AIR_FCST_30_SARIMA_PARAM_2_MODEL_2,
     EXOG_FCST_15_SARIMA_PARAM_EXOG_MODEL_1,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 AIR_TS: pd.DataFrame = load_air_passengers()

kats/tests/models/test_simple_heuristic_model.py

@@ -15,6 +15,8 @@
     SimpleHeuristicModel,
     SimpleHeuristicModelParams,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 test_univariate_df = pd.DataFrame(

kats/tests/models/test_stlf_model.py

@@ -21,6 +21,8 @@
 )
 from kats.models.stlf import STLFModel, STLFParams
 from kats.utils.simulator import Simulator
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_theta_model.py

@@ -28,6 +28,8 @@
     PEYTON_FCST_30_THETA_SM_11,
     PEYTON_FCST_30_THETA_SM_12,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/models/test_var_model.py

@@ -14,6 +14,8 @@
 from kats.consts import TimeSeriesData
 from kats.data.utils import load_data
 from kats.models.var import VARModel, VARParams
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/test_ensemble.py

@@ -29,6 +29,8 @@
 from kats.models.ensemble.median_ensemble import MedianEnsembleModel
 from kats.models.ensemble.weighted_avg_ensemble import WeightedAvgEnsemble
 from kats.models.model import Model
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 np.random.seed(123321)

kats/tests/tsfeatures/test_tsfeatures.py

@@ -25,6 +25,8 @@
     TsFeatures,
     TsFourierFeatures,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 SAMPLE_INPUT_TS_BOCPD_SCALED = pd.DataFrame(

kats/tests/utils/test_datapartition.py

@@ -17,6 +17,8 @@
     SimpleDataPartition,
     SimpleTimestampDataPartition,
 )
+
+# pyre-fixme[21]: Could not find module `parameterized.parameterized`.
 from parameterized.parameterized import parameterized
 
 

kats/tests/utils/test_decomposition.py

@@ -17,6 +17,8 @@
 from kats.utils.decomposition import SeasonalityHandler, TimeSeriesDecomposition
 from kats.utils.simulator import Simulator
 from scipy.stats import ks_2samp
+
+# pyre-fixme[21]: Could not find name `STL` in `statsmodels.tsa.seasonal`.
 from statsmodels.tsa.seasonal import seasonal_decompose, STL
 
 

kats/tests/utils/test_time_series_parameter_tuning.py

@@ -20,6 +20,7 @@
 from kats.models.prophet import ProphetModel
 from kats.utils.time_series_parameter_tuning import compute_search_cardinality
 
+# pyre-fixme[21]: Could not find module `numpy.random`.
 from numpy.random import RandomState
 
 

kats/tsfeatures/tsfeatures.py

@@ -32,6 +32,8 @@
 from scipy.linalg import toeplitz
 from scipy.signal import periodogram  # @manual
 from statsmodels.stats.diagnostic import het_arch
+
+# pyre-fixme[21]: Could not find name `STL` in `statsmodels.tsa.seasonal`.
 from statsmodels.tsa.seasonal import STL
 from statsmodels.tsa.stattools import acf, kpss, pacf
 

kats/utils/decomposition.py

@@ -19,6 +19,8 @@
     ParameterError,
     TimeSeriesData,
 )
+
+# pyre-fixme[21]: Could not find name `STL` in `statsmodels.tsa.seasonal`.
 from statsmodels.tsa.seasonal import seasonal_decompose, STL
 
 # from numpy.typing import ArrayLike