Skip to content

Commit 2ac829f

Browse files
committed
Fix progress bar estimation
1 parent eba5e5f commit 2ac829f

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

src/together/cli/api/utils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,13 +103,13 @@ def generate_progress_bar(
103103
progress = "Progress: [bold red]unavailable[/bold red]"
104104
if finetune_job.status in COMPLETED_STATUSES:
105105
progress = "Progress: [bold green]completed[/bold green]"
106-
elif finetune_job.updated_at is not None:
106+
elif finetune_job.started_at is not None:
107107
# Replace 'Z' with '+00:00' for Python 3.10 compatibility
108-
updated_at_str = finetune_job.updated_at.replace("Z", "+00:00")
109-
update_at = datetime.fromisoformat(updated_at_str).astimezone()
108+
started_at_str = finetune_job.started_at.replace("Z", "+00:00")
109+
started_at = datetime.fromisoformat(started_at_str).astimezone()
110110

111111
if finetune_job.progress is not None:
112-
if current_time < update_at:
112+
if current_time < started_at:
113113
return progress
114114

115115
if not finetune_job.progress.estimate_available:
@@ -118,7 +118,7 @@ def generate_progress_bar(
118118
if finetune_job.progress.seconds_remaining <= 0:
119119
return progress
120120

121-
elapsed_time = (current_time - update_at).total_seconds()
121+
elapsed_time = (current_time - started_at).total_seconds()
122122
ratio_filled = min(
123123
elapsed_time / finetune_job.progress.seconds_remaining, 1.0
124124
)

src/together/types/finetune.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -286,6 +286,7 @@ class FinetuneResponse(BaseModel):
286286
# created/updated datetime stamps
287287
created_at: str | None = None
288288
updated_at: str | None = None
289+
started_at: str | None = None
289290
# job status
290291
status: FinetuneJobStatus | None = None
291292
# job id

0 commit comments

Comments
 (0)