From 3f9909391a5c1e30b91c48b21e7bdc689922de21 Mon Sep 17 00:00:00 2001
From: Landung 'Don' Setiawan <landungs@uw.edu>
Date: Fri, 23 Feb 2024 09:13:42 -0800
Subject: [PATCH] fix: Fix unsafe | operand round 2

---
 src/caustics/models/registry.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/caustics/models/registry.py b/src/caustics/models/registry.py
index 37627da0..97f8a46d 100644
--- a/src/caustics/models/registry.py
+++ b/src/caustics/models/registry.py
@@ -27,10 +27,10 @@ class _KindRegistry(MutableMapping[str, "Parametrized | str"]):
     }
 
     def __init__(self) -> None:
-        self._m: ChainMap[str, Parametrized | str] = ChainMap({}, self.known_kinds)  # type: ignore
+        self._m: ChainMap[str, "Parametrized | str"] = ChainMap({}, self.known_kinds)  # type: ignore
 
     def __getitem__(self, item: str) -> Parametrized:
-        kind_mod: str | Parametrized | None = self._m.get(item, None)
+        kind_mod: "str | Parametrized | None" = self._m.get(item, None)
         if kind_mod is None:
             raise KeyError(f"{item} not in registry")
         if isinstance(kind_mod, str):
@@ -39,7 +39,7 @@ def __getitem__(self, item: str) -> Parametrized:
             cls = getattr(mod, name)  # type: ignore
         return cls
 
-    def __setitem__(self, item: str, value: Parametrized | str) -> None:
+    def __setitem__(self, item: str, value: "Parametrized | str") -> None:
         if not (
             (isinstance(value, type) and issubclass(value, Parametrized))
             or isinstance(value, str)
@@ -71,7 +71,7 @@ def available_kinds() -> list[str]:
 
 def register_kind(
     name: str,
-    cls: Parametrized | str,
+    cls: "Parametrized | str",
     *,
     clobber: bool = False,
 ) -> None:
@@ -96,7 +96,7 @@ def register_kind(
 @lru_cache
 def get_kind(
     name: str,
-) -> Parametrized | None:
+) -> Parametrized:
     """Get a class from the registry by name.
 
     Parameters