fix: mypy in 3.8

neptune-ai · Nov 28, 2024 · 4766351 · 4766351
1 parent 4726ae0
commit 4766351
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/src/neptune_api/proto/descriptor_pool.py b/src/neptune_api/proto/descriptor_pool.py
@@ -1,7 +1,8 @@
+from typing import Optional
 from google.protobuf import descriptor_pool
 
 _DEFAULT_POOL: descriptor_pool.DescriptorPool = descriptor_pool.Default()
-_LOCAL_POOL: descriptor_pool.DescriptorPool | None = None
+_LOCAL_POOL: Optional[descriptor_pool.DescriptorPool] = None
 
 
 if _LOCAL_POOL is None:

diff --git a/src/neptune_retrieval_api/proto/descriptor_pool.py b/src/neptune_retrieval_api/proto/descriptor_pool.py
@@ -1,7 +1,9 @@
+from typing import Optional
+
 from google.protobuf import descriptor_pool
 
 _DEFAULT_POOL: descriptor_pool.DescriptorPool = descriptor_pool.Default()
-_LOCAL_POOL: descriptor_pool.DescriptorPool | None = None
+_LOCAL_POOL: Optional[descriptor_pool.DescriptorPool] = None
 
 
 if _LOCAL_POOL is None:

diff --git a/tests/unit/test_protobuf.py b/tests/unit/test_protobuf.py
@@ -16,12 +16,6 @@
 )
 
 
-def test_hmm():
-    assert list(inspect.getmembers(common_pb2, lambda x: inspect.isclass(x) and issubclass(x, message.Message))) == [
-        "hmm"
-    ]
-
-
 @pytest.mark.parametrize(
     "cls",
     [cls for _, cls in inspect.getmembers(common_pb2, lambda x: inspect.isclass(x) and issubclass(x, message.Message))],