ignore mypy

wandb · Aug 14, 2024 · ab3fa33 · ab3fa33
1 parent 1112450
commit ab3fa33
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/weave/integrations/anthropic/anthropic_sdk.py b/weave/integrations/anthropic/anthropic_sdk.py
@@ -154,15 +154,15 @@ def __stream_text__(self) -> Union[Iterator[str], AsyncIterator[str]]:
         else:
             return self.__sync_stream_text__()
 
-    def __sync_stream_text__(self) -> Iterator[str]:  # type: ignore[attr-defined]
-        for chunk in self:
-            if chunk.type == "content_block_delta" and chunk.delta.type == "text_delta":  # type: ignore[attr-defined]
-                yield chunk.delta.text
-
-    async def __async_stream_text__(self) -> AsyncIterator[str]:  # type: ignore[attr-defined]
-        async for chunk in self:
-            if chunk.type == "content_block_delta" and chunk.delta.type == "text_delta":
-                yield chunk.delta.text
+    def __sync_stream_text__(self) -> Iterator[str]:  # type: ignore
+        for chunk in self:  # type: ignore
+            if chunk.type == "content_block_delta" and chunk.delta.type == "text_delta":  # type: ignore
+                yield chunk.delta.text  # type: ignore
+
+    async def __async_stream_text__(self) -> AsyncIterator[str]:  # type: ignore
+        async for chunk in self:  # type: ignore
+            if chunk.type == "content_block_delta" and chunk.delta.type == "text_delta":  # type: ignore
+                yield chunk.delta.text  # type: ignore
 
     @property
     def text_stream(self) -> Union[Iterator[str], AsyncIterator[str]]: