Skip to content

Commit

Permalink
ignore prev method warnings test case
Browse files Browse the repository at this point in the history
  • Loading branch information
raspawar committed Nov 12, 2024
1 parent 7083c62 commit b115909
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions libs/ai-endpoints/tests/unit_tests/test_structured_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,19 +19,6 @@ class Joke(pydanticV2BaseModel):
rating: Optional[int] = Field(description="How funny the joke is, from 1 to 10")


# def test_method() -> None:
# with pytest.warns(UserWarning) as record:
# with warnings.catch_warnings():
# warnings.filterwarnings(
# "ignore",
# category=UserWarning,
# message=".*not known to support structured output.*",
# )
# ChatNVIDIA(api_key="BOGUS").with_structured_output(Joke, method="json_mode")
# assert len(record) == 1
# assert "unnecessary" in str(record[0].message)


def test_include_raw() -> None:
with pytest.raises(NotImplementedError):
ChatNVIDIA(api_key="BOGUS").with_structured_output(Joke, include_raw=True)
Expand Down

0 comments on commit b115909

Please sign in to comment.