Fix mypy

src/konfetti/cache.py

@@ -1,5 +1,5 @@
 from datetime import datetime, timedelta
-from typing import Dict, Optional, Union
+from typing import Dict, Optional, Union, cast, Any
 
 import attr
 
@@ -91,6 +91,7 @@ def get(self, key):
         data = self._data.get(key, EMPTY)
         if data is EMPTY:
             return data
+        data = cast(Dict[str, Any], data)
         if self._delete_if_expired(key, data["inserted"]):
             return EMPTY
         return data["data"]

src/konfetti/laziness.py

@@ -11,7 +11,7 @@
 @attr.s(slots=True)
 class LazyVariable(DefaultMixin, CastableMixin):
     name = attr.ib(default=NOT_SET)
-    func = attr.ib(default=NOT_SET)
+    func = attr.ib(default=NOT_SET, type=Callable)
     default = attr.ib(default=NOT_SET)
     cast = attr.ib(default=NOT_SET, validator=validate_cast)
 
@@ -38,7 +38,8 @@ def __call__(self, func):
         self.func = func
         name = func.__module__.rsplit(".", 1)[1]
         module = __import__(func.__module__, fromlist=[name])
-        setattr(module, self.name, self)
+        # NOTE. It is better to prevent usage of `lazy` without a passed name
+        setattr(module, self.name, self)  # type: ignore
         return self
 
     def evaluate(self, config):

src/konfetti/vault/base.py

@@ -1,4 +1,4 @@
-from typing import Optional, Union  # ignore: PyUnusedCodeBear
+from typing import Optional, Union, cast  # ignore: PyUnusedCodeBear
 
 import attr
 
@@ -40,12 +40,14 @@ def _get_full_path(self, path):
 
     def _get_from_cache(self, path):
         if self.cache is not NOT_SET:
-            return self.cache.get(path)
+            cache = cast(InMemoryCache, self.cache)
+            return cache.get(path)
         return EMPTY
 
     def _set_to_cache(self, path, value):
         if self.cache is not NOT_SET:
-            self.cache.set(path, value)
+            cache = cast(InMemoryCache, self.cache)
+            cache.set(path, value)
 
     def _get_retry(self, cls, exception):
         retry = self.retry