From 83a427f79085ada941f8a07489afedc2cca7d9c6 Mon Sep 17 00:00:00 2001 From: Doug Davis Date: Mon, 9 Oct 2023 12:57:06 -0500 Subject: [PATCH] Update src/dask_awkward/lib/utils.py Co-authored-by: Angus Hollands --- src/dask_awkward/lib/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/dask_awkward/lib/utils.py b/src/dask_awkward/lib/utils.py index 35d8212c1..e940f2a75 100644 --- a/src/dask_awkward/lib/utils.py +++ b/src/dask_awkward/lib/utils.py @@ -129,8 +129,9 @@ def render_buffer_key(form: Form, form_key: str, attribute: str) -> str: return f"{form_key}-{attribute}" -def parse_buffer_key(buffer_key: str) -> list[str]: - return buffer_key.rsplit("-", maxsplit=1) +def parse_buffer_key(buffer_key: str) -> tuple[str, str]: + head, tail = buffer_key.rsplit("-", maxsplit=1) + return head, tail def form_with_unique_keys(form: Form, key: str) -> Form: