fix test failure

the `for tag in done_tags:` was resetting the existing `tag` variable and breaking things.
replicate · Nov 26, 2024 · 897e28e · 897e28e
1 parent 41adaa9
commit 897e28e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/cog/server/runner.py b/python/cog/server/runner.py
@@ -105,8 +105,8 @@ def predict(
             done_tags = [
                 tag for tag in self._predict_tasks if self._predict_tasks[tag].done()
             ]
-            for tag in done_tags:
-                del self._predict_tasks[tag]
+            for done_tag in done_tags:
+                del self._predict_tasks[done_tag]
 
             self._predict_tasks[tag] = task