MAINT: Update for changes in mypy

Update typing for new errors due to mypy changes
bashtage · Oct 14, 2020 · 8ac812e · 8ac812e
1 parent 7795ac3
commit 8ac812e
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 17 deletions.
diff --git a/arch/bootstrap/base.py b/arch/bootstrap/base.py
@@ -9,6 +9,7 @@
     Sequence,
     Tuple,
     Union,
+    cast,
 )
 
 import numpy as np
@@ -708,8 +709,8 @@ def conf_int(
             values = results
             if method == studentized:
                 # studentized uses studentized parameter estimates
-                assert isinstance(studentized_results, NDArray)
-                values = studentized_results
+                # assert isinstance(studentized_results, NDArray)
+                values = cast(NDArray, studentized_results)
 
             if method in ("debiased", "bc", "bias-corrected", "bca"):
                 # bias corrected uses modified percentiles, but is

diff --git a/arch/unitroot/critical_values/simulation/adf_z_critical_values_simulation_joblib.py b/arch/unitroot/critical_values/simulation/adf_z_critical_values_simulation_joblib.py
@@ -92,9 +92,7 @@ def single_experiment(trend, gen: Generator, file_name: str):
         help="Number of CPUs to use. If not specified, uses cpu_count() - 1",
     )
     parser.add_argument(
-        "--z_only",
-        action="store_true",
-        help="Only execute Z-type tests",
+        "--z_only", action="store_true", help="Only execute Z-type tests",
     )
     args = parser.parse_args()
     njobs = getattr(args, "ncpu", None)

diff --git a/arch/unitroot/critical_values/simulation/phillips-ouliaris-simulation.py b/arch/unitroot/critical_values/simulation/phillips-ouliaris-simulation.py
@@ -316,9 +316,7 @@ def worker(
         help="Number of CPUs to use. If not specified, uses cpu_count() - 1",
     )
     parser.add_argument(
-        "--z_only",
-        action="store_true",
-        help="Only execute Z-type tests",
+        "--z_only", action="store_true", help="Only execute Z-type tests",
     )
     args = parser.parse_args()
     njobs = getattr(args, "ncpu", None)

diff --git a/arch/univariate/volatility.py b/arch/univariate/volatility.py
@@ -2270,7 +2270,7 @@ def compute_variance(
         mus = self._ewma_smoothing_parameters()
 
         sigma2_temp = np.zeros_like(sigma2)
-        assert isinstance(backcast, NDArray)
+        backcast = cast(NDArray, backcast)
         for k in range(kmax):
             mu = mus[k]
             ewma_recursion(mu, resids, sigma2_temp, nobs, backcast[k])
@@ -2348,7 +2348,7 @@ def _simulation_forecast(
         kmax = self.kmax
         w = self._ewma_combination_weights()
         mus = self._ewma_smoothing_parameters()
-        backcast = np.asarray(backcast)
+        backcast = cast(NDArray, np.asarray(backcast))
 
         t = resids.shape[0]
         paths = np.full((t, simulations, horizon), np.nan)
@@ -2359,7 +2359,6 @@ def _simulation_forecast(
         component_one_step = np.empty((kmax, t + 1))
         _resids = np.empty((t + 1))
         _resids[:-1] = resids
-        assert isinstance(backcast, NDArray)
         for k in range(kmax):
             mu = mus[k]
             ewma_recursion(mu, _resids, component_one_step[k, :], t + 1, backcast[k])

diff --git a/arch/utility/testing.py b/arch/utility/testing.py
@@ -26,12 +26,7 @@ class WaldTestStatistic(object):
     """
 
     def __init__(
-        self,
-        stat: float,
-        df: int,
-        null: str,
-        alternative: str,
-        name: str = "",
+        self, stat: float, df: int, null: str, alternative: str, name: str = "",
     ) -> None:
         self._stat = stat
         self._null = null