Skip to content

Commit 30468d0

Browse files
authored
fix(deps): bump datasets versions for all providers (#3382)
Not doing so results in errors of the kind you see in: https://github.com/llamastack/llama-stack-ops/actions/runs/17565092546/job/49890264353
1 parent c9268a7 commit 30468d0

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

llama_stack/providers/registry/datasetio.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def available_providers() -> list[ProviderSpec]:
3030
adapter=AdapterSpec(
3131
adapter_type="huggingface",
3232
pip_packages=[
33-
"datasets",
33+
"datasets>=4.0.0",
3434
],
3535
module="llama_stack.providers.remote.datasetio.huggingface",
3636
config_class="llama_stack.providers.remote.datasetio.huggingface.HuggingfaceDatasetIOConfig",
@@ -42,7 +42,7 @@ def available_providers() -> list[ProviderSpec]:
4242
adapter=AdapterSpec(
4343
adapter_type="nvidia",
4444
pip_packages=[
45-
"datasets",
45+
"datasets>=4.0.0",
4646
],
4747
module="llama_stack.providers.remote.datasetio.nvidia",
4848
config_class="llama_stack.providers.remote.datasetio.nvidia.NvidiaDatasetIOConfig",

llama_stack/providers/registry/post_training.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ def available_providers() -> list[ProviderSpec]:
4848
InlineProviderSpec(
4949
api=Api.post_training,
5050
provider_type="inline::huggingface-gpu",
51-
pip_packages=["trl", "transformers", "peft", "datasets", "torch"],
51+
pip_packages=["trl", "transformers", "peft", "datasets>=4.0.0", "torch"],
5252
module="llama_stack.providers.inline.post_training.huggingface",
5353
config_class="llama_stack.providers.inline.post_training.huggingface.HuggingFacePostTrainingConfig",
5454
api_dependencies=[

0 commit comments

Comments
 (0)