From 13c44f69398a58d43445d9d0a239d13c07f39472 Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Wed, 9 Oct 2024 16:30:01 -0700
Subject: [PATCH] fun

---
 python/tests/unit_tests/test_run_helpers.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/python/tests/unit_tests/test_run_helpers.py b/python/tests/unit_tests/test_run_helpers.py
index 3b19b107..dc5291d5 100644
--- a/python/tests/unit_tests/test_run_helpers.py
+++ b/python/tests/unit_tests/test_run_helpers.py
@@ -7,7 +7,7 @@
 import time
 import uuid
 import warnings
-from typing import Any, AsyncGenerator, Generator, Optional, Set, cast
+from typing import Any, AsyncGenerator, Generator, List, Optional, Set, Tuple, cast
 from unittest.mock import MagicMock, patch
 
 import pytest
@@ -50,7 +50,7 @@ def _get_calls(
     return calls
 
 
-def _get_datas(mock_calls: list[Any]) -> list[tuple[str, dict]]:
+def _get_datas(mock_calls: List[Any]) -> List[Tuple[str, dict]]:
     datas = []
     for call_ in mock_calls:
         data = json.loads(call_.kwargs["data"])
@@ -1509,7 +1509,7 @@ def my_function(a: int) -> Generator[int, None, None]:
 
     with tracing_context(enabled=True):
         with pytest.raises(ValueError, match="foo"):
-            async for _ in my_function(1, langsmith_extra={"client": mock_client}):
+            for _ in my_function(1, langsmith_extra={"client": mock_client}):
                 pass
 
     # Get ALL the call args for the mock_client