Merge pull request #48 from kohei-shinden/fix_s3_client_get_tasks_loc…

…al_cache fix input for local_cache.get in S3Client.get_tasks
m3dev · Feb 1, 2022 · 107e4e2 · 107e4e2
2 parents 322af39 + 9bb4c7e
commit 107e4e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thunderbolt/client/s3_client.py b/thunderbolt/client/s3_client.py
@@ -33,7 +33,7 @@ def get_tasks(self) -> List[Dict[str, Any]]:
             n = n.split('_')
 
             if self.use_cache:
-                cache = self.local_cache.get(x)
+                cache = self.local_cache.get(x['key'])
                 if cache:
                     tasks_list.append(cache)
                     continue
@@ -48,7 +48,7 @@ def get_tasks(self) -> List[Dict[str, Any]]:
                 }
                 tasks_list.append(params)
                 if self.use_cache:
-                    self.local_cache.dump(x, params)
+                    self.local_cache.dump(x['key'], params)
             except Exception:
                 continue