Fixing tests to reflect NotRequired

blast-hardcheese · blast-hardcheese · commit 48bcf1e958de · 2025-01-31T15:56:28.000-08:00
diff --git a/tests/codegen/rpc/generated/test_service/rpc_method.py b/tests/codegen/rpc/generated/test_service/rpc_method.py
@@ -10,6 +10,7 @@
     Literal,
     Optional,
     Mapping,
+    NotRequired,
     Union,
     Tuple,
     TypedDict,
diff --git a/tests/codegen/stream/generated/test_service/stream_method.py b/tests/codegen/stream/generated/test_service/stream_method.py
@@ -10,6 +10,7 @@
     Literal,
     Optional,
     Mapping,
+    NotRequired,
     Union,
     Tuple,
     TypedDict,
