Merge pull request #773 from bashtage/examples-typing

TYP: Small typing improvements

Author: bashtage

examples/multiple-comparison_examples.ipynb

@@ -332,7 +332,7 @@
     "better_models = pd.concat(\n",
     "    [model_losses_df.mean(axis=0), model_losses_df.mean(axis=0)], axis=1\n",
     ")\n",
-    "better_models.columns = [\"Same or worse\", \"Better\"]\n",
+    "better_models.columns = pd.Index([\"Same or worse\", \"Better\"])\n",
     "better = better_models.index.isin(stepm.superior_models)\n",
     "worse = np.logical_not(better)\n",
     "better_models.loc[better, \"Same or worse\"] = np.nan\n",

examples/unitroot_examples.ipynb

@@ -204,7 +204,7 @@
     "\n",
     "resids = pd.DataFrame(reg_res.resid)\n",
     "resids.index = default.index[6:]\n",
-    "resids.columns = [\"resids\"]\n",
+    "resids.columns = pd.Index([\"resids\"])\n",
     "fig = resids.plot()"
    ]
   },

examples/univariate_volatility_scenarios.ipynb

@@ -168,7 +168,9 @@
     "random_state = np.random.RandomState(1)\n",
     "\n",
     "\n",
-    "def scenario_rng(size):\n",
+    "def scenario_rng(\n",
+    "    size: tuple[int, int],\n",
+    ") -> np.ndarray[tuple[int, int], np.dtype[np.float64]]:\n",
     "    shocks = random_state.standard_normal(size)\n",
     "    shocks[:, :5] *= np.sqrt(2)\n",
     "    return shocks\n",
@@ -197,13 +199,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "class ScenarioBootstrapRNG(object):\n",
-    "    def __init__(self, shocks, random_state):\n",
-    "        self._shocks = np.asarray(shocks)  # 1d\n",
+    "class ScenarioBootstrapRNG:\n",
+    "    def __init__(\n",
+    "        self,\n",
+    "        shocks: np.ndarray[tuple[int], np.dtype[np.float64]],\n",
+    "        random_state: np.random.RandomState,\n",
+    "    ) -> None:\n",
+    "        self._shocks = shocks  # 1d\n",
     "        self._rs = random_state\n",
     "        self.n = shocks.shape[0]\n",
     "\n",
-    "    def rng(self, size):\n",
+    "    def rng(self, size: int) -> np.ndarray[tuple[int], np.dtype[np.float64]]:\n",
     "        idx = self._rs.randint(0, self.n, size=size)\n",
     "        return self._shocks[idx]\n",
     "\n",