worker: fix lint

pgq · Jul 11, 2022 · 22cb2eb · 22cb2eb
1 parent 92d5711
commit 22cb2eb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pgq/cascade/worker.py b/pgq/cascade/worker.py
@@ -182,7 +182,10 @@ def is_batch_done(self, state: DictRow, batch_info: BatchInfo, dst_db: Connectio
         # on combined-branch the target can get several batches ahead
         if self._worker_state and self._worker_state.wait_behind:
             self.wait_for_tick(dst_db, batch_info['tick_id'])
-            state = self._consumer_state
+
+            # refresh state
+            if self._consumer_state:
+                state = self._consumer_state
 
         # handle combined_queue type change (branch->root)
         if self._worker_state and self.was_wait_behind: