From ac0d99f2a509d06f94a6b5ade21c8732e1f222cb Mon Sep 17 00:00:00 2001
From: Will Fatherley <wfatherley@proton.me>
Date: Sat, 3 Aug 2024 22:46:43 -0400
Subject: [PATCH] ensure hkdf can return entirety of expand

---
 src/hkdfref/_hkdfref.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/hkdfref/_hkdfref.py b/src/hkdfref/_hkdfref.py
index 0ca188a..4fab850 100644
--- a/src/hkdfref/_hkdfref.py
+++ b/src/hkdfref/_hkdfref.py
@@ -7,7 +7,7 @@
 from typing import Any, Callable
 
 
-logger = getLogger(__name__)
+logger = getLogger(__name__[1:])
 
 
 default_hash = hashlib.sha512
@@ -39,7 +39,9 @@ def hkdf(*ikm: bytes, **kwargs: Any) -> bytes:
         prk = extract(salt, *ikm, hash=extract_hash)
     else:
         prk = b"".join(ikm)
-    return expand(prk, info, l, expand_hash)
+    return expand(
+        prk, info, l, expand_hash, return_all=kwargs.get("return_all", False)
+    )
 
 
 def expand(