From f6c62725ede84604992dd7c93c377357e915c91f Mon Sep 17 00:00:00 2001
From: MSchmatzAISI <Michael.Schmatz@dsit.gov.uk>
Date: Tue, 1 Oct 2024 14:37:55 +0100
Subject: [PATCH] Fix hf dataset type bug (#596)

Co-authored-by: jjallaire-aisi <joseph.allaire@dsit.gov.uk>
---
 src/inspect_ai/dataset/_sources/hf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/inspect_ai/dataset/_sources/hf.py b/src/inspect_ai/dataset/_sources/hf.py
index 072587182..092c35960 100644
--- a/src/inspect_ai/dataset/_sources/hf.py
+++ b/src/inspect_ai/dataset/_sources/hf.py
@@ -25,7 +25,7 @@ def hf_dataset(
     seed: int | None = None,
     limit: int | None = None,
     trust: bool = False,
-    **kwargs: dict[str, Any],
+    **kwargs: Any,
 ) -> Dataset:
     """Datasets read using the Hugging Face `datasets` package.