Skip to content

Commit

Permalink
Add job logs to inference test
Browse files Browse the repository at this point in the history
  • Loading branch information
javiermtorres committed Dec 12, 2024
1 parent 7e02ab1 commit 86b8dc6
Showing 1 changed file with 10 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
from lumigator_schemas.datasets import DatasetFormat, DatasetResponse
from lumigator_schemas.experiments import ExperimentResponse
from lumigator_schemas.extras import ListingResponse
from lumigator_schemas.jobs import JobResponse, JobStatus
from lumigator_schemas.jobs import JobLogsResponse, JobResponse, JobStatus

from backend.main import app

Expand Down Expand Up @@ -108,6 +108,15 @@ def test_upload_data_launch_job(
succeeded = False
break
time.sleep(30)

logs_infer_job_response = local_client.get(
f"/health/jobs/{create_inference_job_response_model.id}/logs"
)
logs_infer_job_response_model = JobLogsResponse.model_validate(logs_infer_job_response.json())
print(f"-- logs -- {create_inference_job_response_model.id}")
print(logs_infer_job_response_model.logs)
print("----------")

assert succeeded


Expand Down

0 comments on commit 86b8dc6

Please sign in to comment.