Skip to content

Commit

Permalink
Filter out cohere's pydantic warning (#1081)
Browse files Browse the repository at this point in the history
  • Loading branch information
collindutter authored Aug 19, 2024
1 parent 49495dd commit ade80a5
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions griptape/drivers/prompt/cohere_prompt_driver.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from __future__ import annotations

import warnings
from typing import TYPE_CHECKING, Any

from attrs import Factory, define, field
Expand All @@ -24,6 +25,9 @@
from griptape.tokenizers import BaseTokenizer, CohereTokenizer
from griptape.utils import import_optional_dependency

# TODO Remove once https://github.com/cohere-ai/cohere-python/issues/559 is resolved
warnings.filterwarnings("ignore", module="pydantic")

if TYPE_CHECKING:
from collections.abc import Iterator

Expand Down

0 comments on commit ade80a5

Please sign in to comment.