Skip to content

Commit 49c27e7

Browse files
committed
fix: update for mypy lint errors
1 parent b2e3700 commit 49c27e7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/joserfc/_rfc8037/okp_key.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -94,8 +94,11 @@ class OKPKey(CurveKey[PrivateOKPKey, PublicOKPKey]):
9494
def exchange_derive_key(self, key: "OKPKey") -> bytes:
9595
# used in ECDH-ES Algorithms
9696
pubkey: t.Union[X25519PublicKey, X448PublicKey] = key.get_op_key("deriveKey") # type: ignore[assignment]
97-
if ((isinstance(self.private_key, X25519PrivateKey) and isinstance(pubkey, X25519PublicKey)) or
98-
(isinstance(self.private_key, X448PrivateKey) and isinstance(pubkey, X448PublicKey))):
97+
98+
# this if else logic is used for type hints
99+
if isinstance(self.private_key, X25519PrivateKey) and isinstance(pubkey, X25519PublicKey):
100+
return self.private_key.exchange(pubkey)
101+
elif isinstance(self.private_key, X448PrivateKey) and isinstance(pubkey, X448PublicKey):
99102
return self.private_key.exchange(pubkey)
100103
raise InvalidExchangeKeyError()
101104

0 commit comments

Comments
 (0)