Fix type annotations for add_key (#101)

Nitrokey · Nov 23, 2023 · 61b742c · 61b742c
1 parent 1eb39be
commit 61b742c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/nethsm/__init__.py b/nethsm/__init__.py
@@ -888,7 +888,7 @@ def get_key_public_key(self, key_id: str) -> str:
 
     def add_key(
         self,
-        key_id: str,
+        key_id: Optional[str],
         type: KeyType,
         mechanisms: list[KeyMechanism],
         private_key: PrivateKey,
@@ -954,6 +954,7 @@ def add_key(
                     409: f"Conflict -- a key with the ID {key_id} already exists",
                 },
             )
+        assert key_id
         return key_id
 
     def delete_key(self, key_id: str) -> None: