fix: flake8 and black errors

provos · Sep 3, 2024 · 88dbf7f · 88dbf7f
1 parent a339573
commit 88dbf7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/src/planai/cached_task.py b/src/planai/cached_task.py
@@ -105,4 +105,4 @@ def _cache_up_call(self, input_task: Task, cached_results: List[Task]):
                 self._cache.set(cache_key, cached_results)
             logging.info("Task %s cached results for key: %s", self.name, cache_key)
         except Exception as e:
-            logging.error("Error caching results for key %s: %s", cache_key, str(e))
+            logging.error("Error caching results for key %s: %s", cache_key, str(e))
diff --git a/src/planai/task.py b/src/planai/task.py
@@ -100,10 +100,6 @@ def prefix_for_input_task(
         return None
 
 
-import logging
-import threading
-
-
 class WorkBufferContext:
     def __init__(self, worker, input_task=None):
         self.worker: "TaskWorker" = worker
@@ -133,7 +129,7 @@ def _flush_work_buffer(self):
                 self.worker._dispatch_work(task)
         self.work_buffer.clear()
 
-    def add_to_buffer(self, consumer: 'TaskWorker', task: "Task"):
+    def add_to_buffer(self, consumer: "TaskWorker", task: "Task"):
         self.work_buffer.append((consumer, task))