xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-07 14:47:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: PollingOperation progress bar update progress by absolute value (#65)

Browse Source
This commit is contained in:
Christian Byrne 2025-04-30 09:09:27 -07:00 committed by GitHub
parent 1bae2de0d8
commit 021e64db6c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy_api_nodes/apis/client.py
View File

@ -117,6 +117,8 @@ T = TypeVar("T", bound=BaseModel)
R = TypeVar("R", bound=BaseModel) R = TypeVar("R", bound=BaseModel)
P = TypeVar("P", bound=BaseModel) # For poll response P = TypeVar("P", bound=BaseModel) # For poll response
PROGRESS_BAR_MAX = 100
class EmptyRequest(BaseModel): class EmptyRequest(BaseModel):
"""Base class for empty request bodies. """Base class for empty request bodies.
@ -508,7 +510,7 @@ class PollingOperation(Generic[T, R]):
"""Poll until the task is complete""" """Poll until the task is complete"""
poll_count = 0 poll_count = 0
if self.progress_extractor: if self.progress_extractor:
progress = utils.ProgressBar(100) progress = utils.ProgressBar(PROGRESS_BAR_MAX)
while True: while True:
try: try:
@ -547,7 +549,7 @@ class PollingOperation(Generic[T, R]):
if self.progress_extractor: if self.progress_extractor:
new_progress = self.progress_extractor(response_obj) new_progress = self.progress_extractor(response_obj)
if new_progress is not None: if new_progress is not None:
progress.update(new_progress) progress.update_absolute(new_progress, total=PROGRESS_BAR_MAX)
if status == TaskStatus.COMPLETED: if status == TaskStatus.COMPLETED:
logging.debug("[DEBUG] Task completed successfully") logging.debug("[DEBUG] Task completed successfully")