Merge pull request #28 from syrusakbary/get-cache-key-override

Support for overriding get_cache_key in subclasses
syrusakbary · Nov 14, 2022 · b07ca57 · b07ca57
2 parents 3b9d8bf + 3672ac3
commit b07ca57
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/aiodataloader.py b/aiodataloader.py
@@ -82,7 +82,10 @@ def __init__(
         if cache is not None:
             self.cache = cache
 
-        self.get_cache_key = get_cache_key or (lambda x: x)
+        if get_cache_key is not None:
+            self.get_cache_key = get_cache_key
+        if not hasattr(self, 'get_cache_key'):
+            self.get_cache_key = lambda x: x
 
         self._cache = cache_map if cache_map is not None else {}
         self._queue: List[Loader] = []