fix tests

jpadilla · Sep 12, 2024 · 4d45193 · 4d45193
1 parent 95c3004
commit 4d45193
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jwt/algorithms.py b/jwt/algorithms.py
@@ -770,8 +770,9 @@ def __init__(self, **kwargs: Any) -> None:
             pass
 
         def prepare_key(self, key: AllowedOKPKeys | str | bytes) -> AllowedOKPKeys:
-            if isinstance(key, self._crypto_key_types):
-                return key
+            if not isinstance(key, (str, bytes)):
+                self.check_crypto_key_type(key)
+                return cast("AllowedOKPKeys", key)
 
             key_str = key.decode("utf-8") if isinstance(key, bytes) else key
             key_bytes = key.encode("utf-8") if isinstance(key, str) else key