Skip to content

Commit

Permalink
skip summarization tests as well
Browse files Browse the repository at this point in the history
  • Loading branch information
fgebhart committed Dec 4, 2024
1 parent ffdebc4 commit f76498b
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 4 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/integration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -27,4 +27,4 @@ jobs:
poetry run pytest
env:
TEST_API_URL: ${{ secrets.TEST_API_URL }}
TEST_TOKEN: ${{ secrets.AA_API_TOKEN }}
TEST_TOKEN: ${{ secrets.TEST_TOKEN }}
5 changes: 2 additions & 3 deletions tests/test_prompt.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,10 +140,9 @@ def test_image_controls_with_cats_and_dogs(sync_client: Client):
text = Text.from_text("A picture of ")
request = CompletionRequest(
prompt=Prompt([image, text]),
model="pharia-1-llm-7b-control",
maximum_tokens=16,
control_log_additive=True,
disable_optimizations=False,
)
result = sync_client.complete(request, model="luminous-extended")
assert result.completions[0].completion == " a dog"
result = sync_client.complete(request, model="luminous-base")
assert result.completions[0].completion == " a dog with a blank sign"
6 changes: 6 additions & 0 deletions tests/test_summarize.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,10 @@
# AsyncClient


SKIP_REASON = "skipping as summarization endpoint is deprecated and required model is no longer available"


@pytest.mark.skip(reason=SKIP_REASON)
@pytest.mark.system_test
async def test_can_summarize_with_async_client(async_client: AsyncClient):
request = SummarizationRequest(
Expand All @@ -24,6 +28,7 @@ async def test_can_summarize_with_async_client(async_client: AsyncClient):
# Client


@pytest.mark.skip(reason=SKIP_REASON)
@pytest.mark.system_test
def test_summarize(sync_client: Client):
# when posting a Summarization request
Expand All @@ -37,6 +42,7 @@ def test_summarize(sync_client: Client):
assert response.summary is not None


@pytest.mark.skip(reason=SKIP_REASON)
@pytest.mark.system_test
def test_text(sync_client: Client):
request = SummarizationRequest(
Expand Down

0 comments on commit f76498b

Please sign in to comment.