Revert "Filter out cohere's pydantic warning (#1081)" (#1147)

griptape-ai · Sep 5, 2024 · 21c9d21 · 21c9d21
1 parent a1ad5b7
commit 21c9d21
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/griptape/drivers/prompt/cohere_prompt_driver.py b/griptape/drivers/prompt/cohere_prompt_driver.py
@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import warnings
 from typing import TYPE_CHECKING, Any
 
 from attrs import Factory, define, field
@@ -25,9 +24,6 @@
 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
 

diff --git a/poetry.lock b/poetry.lock