Skip to content

Commit 50168d6

Browse files
committed
fix tests
1 parent 2ac829f commit 50168d6

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/unit/test_cli_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
def create_finetune_response(
1414
status: FinetuneJobStatus = FinetuneJobStatus.STATUS_RUNNING,
15-
updated_at: str = "2024-01-01T12:00:00Z",
15+
created_at: str = "2024-01-01T12:00:00Z",
1616
progress: FinetuneProgress | None = None,
1717
job_id: str = "ft-test-123",
1818
) -> FinetuneResponse:
@@ -30,7 +30,7 @@ def create_finetune_response(
3030
return FinetuneResponse(
3131
id=job_id,
3232
progress=progress,
33-
updated_at=updated_at,
33+
created_at=created_at,
3434
status=status,
3535
)
3636

@@ -359,7 +359,7 @@ def test_timezone_aware_datetime(self):
359359
"""Test with different timezone for updated_at."""
360360
current_time = datetime(2024, 1, 1, 12, 0, 30, tzinfo=timezone.utc)
361361
finetune_job = create_finetune_response(
362-
updated_at="2024-01-01T07:00:00-05:00", # Same as 12:00:00 UTC (EST = UTC-5)
362+
created_at="2024-01-01T07:00:00-05:00", # Same as 12:00:00 UTC (EST = UTC-5)
363363
progress=FinetuneProgress(estimate_available=True, seconds_remaining=60.0),
364364
)
365365

@@ -385,7 +385,7 @@ def test_negative_elapsed_time_scenario(self):
385385
"""Test unusual case where current time appears before updated_at."""
386386
current_time = datetime(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc)
387387
finetune_job = create_finetune_response(
388-
updated_at="2024-01-01T12:00:30Z", # In the "future"
388+
created_at="2024-01-01T12:00:30Z", # In the "future"
389389
progress=FinetuneProgress(estimate_available=True, seconds_remaining=100.0),
390390
)
391391

0 commit comments

Comments
 (0)