From 8a7f509a4d0de11af93cf183ee0a4604cc87af70 Mon Sep 17 00:00:00 2001
From: Albert Hsiung <ahsiung@seatgeek.com>
Date: Mon, 31 Oct 2022 13:56:16 -0400
Subject: [PATCH] more accurate type hints

---
 .pre-commit-config.yaml  | 4 ++--
 cmdstanpy/stanfit/mle.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 0fe686b9..42864c8a 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -12,7 +12,7 @@ repos:
       - id: isort
   # https://github.com/python/black#version-control-integration
   - repo: https://github.com/psf/black
-    rev: 21.6b0
+    rev: 22.10.0
     hooks:
       - id: black
   - repo: https://github.com/pycqa/flake8
@@ -20,7 +20,7 @@ repos:
     hooks:
       - id: flake8
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v0.910
+    rev: v0.982
     hooks:
       - id: mypy
         # Copied from setup.cfg
diff --git a/cmdstanpy/stanfit/mle.py b/cmdstanpy/stanfit/mle.py
index 303d831d..ffdf3fed 100644
--- a/cmdstanpy/stanfit/mle.py
+++ b/cmdstanpy/stanfit/mle.py
@@ -152,7 +152,7 @@ def optimized_iterations_pd(self) -> Optional[pd.DataFrame]:
         return pd.DataFrame(self._all_iters, columns=self.column_names)
 
     @property
-    def optimized_params_dict(self) -> Dict[str, float]:
+    def optimized_params_dict(self) -> Dict[str, np.float64]:
         """
         Returns all estimates from the optimizer, including `lp__` as a
         Python Dict.  Only returns estimate from final iteration.