commit b545db0453894b66896e0432687ce8c7abea4f38
author Dmitri Prime <bolms@google.com> Thu Sep 19 15:00:13 2024 -0700
committer GitHub <noreply@github.com> Thu Sep 19 15:00:13 2024 -0700
tree cb759939cb910c760c9b8646542f10ba77bfc270
parent cffcf47c39f83332c55efd0b8ad1216c8f28f8ec

Fix minor type annotation lints. (#183)

compiler/util/ir_data_fields.py

diff --git a/compiler/util/ir_data_fields.py b/compiler/util/ir_data_fields.py
index 002ea3b..af6fbda 100644
--- a/compiler/util/ir_data_fields.py
+++ b/compiler/util/ir_data_fields.py
@@ -79,7 +79,7 @@
     """A list that makes copies of any value that is inserted"""
 
     def __init__(
-        self, value_type: CopyValuesListT, iterable: Optional[Iterable] = None
+        self, value_type: CopyValuesListT, iterable: Optional[Iterable[Any]] = None
     ):
         if iterable:
             super().__init__(iterable)

compiler/util/ir_data_utils.py

diff --git a/compiler/util/ir_data_utils.py b/compiler/util/ir_data_utils.py
index 2d38eac..8347625 100644
--- a/compiler/util/ir_data_utils.py
+++ b/compiler/util/ir_data_utils.py
@@ -322,7 +322,7 @@
     return ir_data_fields.build_default(spec)
 
 
-def _field_type(ir_or_spec: Union[MessageT, ir_data_fields.FieldSpec]) -> type:
+def _field_type(ir_or_spec: Union[MessageT, ir_data_fields.FieldSpec]) -> type[Any]:
     if isinstance(ir_or_spec, ir_data_fields.FieldSpec):
         return ir_or_spec.data_type
     return type(ir_or_spec)