From fd6b684f33bba099039d4eec68b35f2be2235a58 Mon Sep 17 00:00:00 2001 From: James Darpinian Date: Wed, 2 Sep 2026 09:13:10 -0700 Subject: [PATCH] Fix imagenette dataset loading with newer datasets versions datasets>=4 dropped support for script-based HF dataset repos, which breaks frgfm/imagenette (used by inet10.py). Pin datasets<4 and pass trust_remote_code=True plus the required config name. Co-Authored-By: Claude Sonnet 5 Claude-Session: https://claude.ai/code/session_01WPB1n5wdQgds2B826U9jDa --- inet10.py | 2 ++ pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/inet10.py b/inet10.py index ad93b2c..5c6b7de 100644 --- a/inet10.py +++ b/inet10.py @@ -27,8 +27,10 @@ def crop(): transform = transforms.MultiViewTransform({f"view_{i}": crop() for i in range(views)}) dataset = spt.data.HFDataset( "frgfm/imagenette", + "320px", split=split, transform=transform, + trust_remote_code=True, ) return torch.utils.data.DataLoader( dataset, diff --git a/pyproject.toml b/pyproject.toml index d680700..6169e96 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -2,7 +2,7 @@ name = "lejepa-tutorials" version = "0.1.0" requires-python = ">=3.10" -dependencies = ["stable-pretraining"] +dependencies = ["stable-pretraining", "datasets<4"] [tool.setuptools] py-modules = ["app", "debug_lewm", "inet10", "mmnist", "utils"]