Skip to content

Commit

Permalink
test call model url directly (#163)
Browse files Browse the repository at this point in the history
  • Loading branch information
sambles committed Nov 17, 2023
1 parent 7811cdc commit 157c82d
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion tests/modelcheck.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,11 @@ def wait_for_api(module_scoped_container_getter):
# Wait for Model
oasis_client = APIClient(api_url=api_url)
oasis_client.api.mount('http://', HTTPAdapter(max_retries=retries))
assert oasis_client.models.get(1)

model_headers = {'authorization': f"Bearer {oasis_client.api.tkn_access}"}
model_url = f"{api_url}/V1/models/1/"
assert request_session.get(model_url, headers=model_headers)
#assert oasis_client.models.get(1)
return oasis_client


Expand Down

0 comments on commit 157c82d

Please sign in to comment.