Merge pull request #631 from ahsiungsg/more_accurate_type_hints

more accurate type hints

Author: WardBrian

.pre-commit-config.yaml

@@ -12,15 +12,15 @@ 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
     rev: 3.9.2
     hooks:
       - id: flake8
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v0.910
+    rev: v0.982
     hooks:
       - id: mypy
         # Copied from setup.cfg

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.