Skip to content

Commit f6d9a20

Browse files
committed
Fix type check
1 parent 0fba07f commit f6d9a20

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

pycardano/crypto/bip32.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -326,7 +326,7 @@ def derive(
326326
index += 2**31
327327

328328
if private:
329-
if not self._xprivate_key:
329+
if self._xprivate_key is None:
330330
raise ValueError("Missing private key. Can't do private derivation.")
331331

332332
private_node = (
@@ -336,17 +336,22 @@ def derive(
336336
self._chain_code,
337337
self._path,
338338
)
339-
return self._derive_private_child_key_by_index(private_node, index)
340339

341-
if not self._public_key:
342-
raise ValueError("Missing public key. Can't do public derivation.")
340+
if any(x is None for x in private_node):
341+
raise ValueError(f"None values in private node: {private_node}")
342+
343+
return self._derive_private_child_key_by_index(private_node, index) # type: ignore
343344

344345
public_node = (
345346
self._public_key,
346347
self._chain_code,
347348
self._path,
348349
)
349-
return self._derive_public_child_key_by_index(public_node, index)
350+
351+
if any(x is None for x in public_node):
352+
raise ValueError(f"None values in public node: {public_node}")
353+
354+
return self._derive_public_child_key_by_index(public_node, index) # type: ignore
350355

351356
def _derive_private_child_key_by_index(
352357
self, private_pnode: Tuple[bytes, bytes, bytes, bytes, str], index: int

0 commit comments

Comments
 (0)