From 7b3c11bb286300c0ec0ec75d8fc529f22d171c6d Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Mon, 12 May 2025 02:20:09 +0000
Subject: [PATCH] Update datasets requirement from ~=3.2.0 to >=3.2,<3.7

Updates the requirements on [datasets](https://github.com/huggingface/datasets) to permit the latest version.
- [Release notes](https://github.com/huggingface/datasets/releases)
- [Commits](https://github.com/huggingface/datasets/compare/3.2.0...3.6.0)

---
updated-dependencies:
- dependency-name: datasets
  dependency-version: 3.6.0
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
---
 pyproject.toml   | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 13bcee2..81dc8a9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -48,7 +48,7 @@ dependencies = [ #TODO: can add optional dependencies
 'torch-geometric~=2.5.2',
 'torch-scatter~=2.1.2',
 'transformers~=4.42.4',
-'datasets~=3.2.0', #for HF datasets
+'datasets>=3.2,<3.7', #for HF datasets
 # 'causal-conv1d>=1.4.0', #For jamba/mamba
 # 'mamba-ssm~=2.2.4', #For jamba/mamba
 # 'flash-attn~=2.7.4.post1', #Flash Attention
diff --git a/requirements.txt b/requirements.txt
index be402e0..33d8af3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -14,7 +14,7 @@ scipy~=1.15.2
 torch-geometric~=2.5.2
 torch-scatter~=2.1.2
 transformers~=4.42.4
-datasets~=3.2.0 #for HF datasets
+datasets>=3.2,<3.7 #for HF datasets
 # causal-conv1d>=1.4.0 #For jamba/mamba
 # mamba-ssm~=2.2.4 #For jamba/mamba
 # flash-attn~=2.7.4.post1 #Flash Attention