HSMAUTH: add and change type annotations

Yubico · Aug 9, 2023 · 58ce91e · 58ce91e
1 parent 0b6bea9
commit 58ce91e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yubikit/hsmauth.py b/yubikit/hsmauth.py
@@ -194,7 +194,7 @@ class SessionKeys(NamedTuple):
     key_srmac: bytes
 
     @classmethod
-    def parse(cls, response) -> "SessionKeys":
+    def parse(cls, response: bytes) -> "SessionKeys":
         key_senc = response[:16]
         key_smac = response[16:32]
         key_srmac = response[32:48]
@@ -318,7 +318,7 @@ def put_credential_derived(
         management_key: bytes,
         label: str,
         credential_password: Union[bytes, str],
-        derivation_password: Union[bytes, str],
+        derivation_password: str,
         touch_required: bool = False,
     ) -> Credential:
         """Import a symmetric YubiHSM Auth credential derived from password"""