Fixes for mypy 0.990 (#672)

SeldonIO · Nov 8, 2022 · 42c5f16 · 42c5f16
1 parent 529e1a4
commit 42c5f16
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -4,7 +4,7 @@ repos:
     hooks:
       - id: flake8
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v0.942
+    rev: v0.990
     hooks:
       - id: mypy
         additional_dependencies: [

diff --git a/alibi_detect/base.py b/alibi_detect/base.py
@@ -169,7 +169,7 @@ def _set_config(self, inputs):  # TODO - move to BaseDetector once config save/l
         name = self.__class__.__name__
 
         # Init config dict
-        self.config: Dict[str, Any] = {
+        self.config = {
             'name': name,
             'meta': {
                 'version': __version__,

diff --git a/alibi_detect/utils/keops/kernels.py b/alibi_detect/utils/keops/kernels.py
@@ -57,7 +57,7 @@ class GaussianRBF(nn.Module):
     def __init__(
         self,
         sigma: Optional[torch.Tensor] = None,
-        init_sigma_fn: Callable = None,
+        init_sigma_fn: Optional[Callable] = None,
         trainable: bool = False
     ) -> None:
         """