more typing annotations

paradigmxyz · Jan 28, 2024 · 0f8cb22 · 0f8cb22
1 parent ff59205
commit 0f8cb22
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_mesc.py b/tests/test_mesc.py
@@ -14,19 +14,20 @@
     Generator,
     Mapping,
     cast,
+    Union,
 )
 import pytest
 
 import generate
-from mesc.types import EndpointQuery, MultiEndpointQuery, RpcConfig
+from mesc.types import RpcConfig, MescQuery
 
 
 # tests are in form [test_name, env, config, query, result, should_succeed]
 Test = tuple[
     str,
     dict[str, str],
     RpcConfig,
-    None | EndpointQuery | MultiEndpointQuery,
+    Union[None, MescQuery],
     Any,
     bool,
 ]