From de7b722789475e1359ebd33ce01eea71739ef7b3 Mon Sep 17 00:00:00 2001 From: Milo Cress Date: Mon, 30 Sep 2024 17:31:00 -0400 Subject: [PATCH] oops --- llmfoundry/data/finetuning/tasks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llmfoundry/data/finetuning/tasks.py b/llmfoundry/data/finetuning/tasks.py index ec170873d2..185a6c81b8 100644 --- a/llmfoundry/data/finetuning/tasks.py +++ b/llmfoundry/data/finetuning/tasks.py @@ -170,7 +170,7 @@ def _is_empty_or_nonexistent(dirpath: str) -> bool: return not os.path.isdir(dirpath) or len(os.listdir(dirpath)) == 0 -def _get_key(dictionary: Mapping[str, Any], allowed_keys: set[str], error_type): +def _get_key(dictionary: Mapping[str, Any], allowed_keys: set[str]): if not isinstance(dictionary, Mapping): raise InvalidExampleTypeError(str(type(dictionary))) desired_keys = allowed_keys.intersection(dictionary.keys())