From c50da898dcb5ea3566c051b178aaa372aca4b7b6 Mon Sep 17 00:00:00 2001 From: Vicki Boykis Date: Mon, 16 Sep 2024 14:27:21 -0400 Subject: [PATCH] hard pin datasets and wandb --- pyproject.toml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 6bfd9a5..4c5f074 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,7 +23,7 @@ dependencies = [ "click>=8.1.7", "scipy==1.13.1", "numpy<2.0.0", - "wandb>=0.16.3", + "wandb==0.16.3", "protobuf>=3.20.2", "urllib3>=1.26.18,<2", "pydantic>=2.6.4", @@ -38,7 +38,7 @@ ruff = ["ruff==0.5.5"] jobs = [ # HuggingFace / pytorch "torch==2.4.0", - "datasets>=2.20.0", + "datasets==2.20.0", "transformers==4.43.4", "accelerate==0.33.0", "peft==0.12.0",