From 213873f1b26c0edd28722c0bab3156a8a45973e1 Mon Sep 17 00:00:00 2001
From: Glenn Waters <glenn@watrs.ca>
Date: Fri, 31 Jan 2025 15:50:29 -0500
Subject: [PATCH] More types.

---
 upb_lib/elements.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/upb_lib/elements.py b/upb_lib/elements.py
index 0179628..8dbdf08 100644
--- a/upb_lib/elements.py
+++ b/upb_lib/elements.py
@@ -89,7 +89,9 @@ def _notify(self) -> None:
             observer(self, self._changeset)
         self._changeset = {}
 
-    def setattr(self, attr, new_value, close_the_changeset=True):
+    def setattr(
+        self, attr: str, new_value: Any, close_the_changeset: bool = True
+    ) -> None:
         """If attribute value has changed then set it and call the callbacks"""
         existing_value = getattr(self, attr, None)
         if existing_value != new_value:
@@ -125,7 +127,7 @@ def __init__(self, pim):
         self.pim = pim
         self.elements: dict[str, T] = {}
 
-    def add_element(self, element):
+    def add_element(self, element: T):
         """Add an element to list of elements."""
         self.elements[element.index] = element
 
@@ -134,7 +136,7 @@ def connection_status_change(self, _):
         for _, element in self.elements.items():
             element._notify()
 
-    def sync(self):
+    def sync(self) -> None:
         """Should be overridden by derived class."""
         raise NotImplementedError()