From 86959e3577fa2c1c92c96d4ca0aa39b497a14a23 Mon Sep 17 00:00:00 2001
From: Nicolas Rabault <rabault.nicolas@gmail.com>
Date: Mon, 29 Jun 2026 14:28:43 +0200
Subject: [PATCH] test(jobs): skip test_hf if datasets extra is missing

lerobot.configs.train pulls in datasets at import time, so the module
fails to collect without lerobot[dataset]. Guard with importorskip,
matching the convention in tests/training/test_multi_gpu.py.
---
 tests/jobs/test_hf.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/jobs/test_hf.py b/tests/jobs/test_hf.py
index 90ebd9c67..3b275cb95 100644
--- a/tests/jobs/test_hf.py
+++ b/tests/jobs/test_hf.py
@@ -21,6 +21,8 @@ import draccus
 import httpx
 import pytest
 
+pytest.importorskip("datasets", reason="datasets is required (install lerobot[dataset])")
+
 from lerobot.configs.train import TrainPipelineConfig
 from lerobot.jobs.hf import (
     _pod_forwarded_args,