chore: huggingface-hub is optional requirement

Author: bpiwowar

pyproject.toml

@@ -36,7 +36,6 @@ dependencies = [
     "httpx >=0.26,<1",
     "humanfriendly >=10",
     "jinja2 >=3",
-    "huggingface-hub >0.17",
     "marshmallow >=3.20,<4",
     "omegaconf >=2.3,<3",
     "psutil >=7,<8",
@@ -74,6 +73,9 @@ docs = [
     "sphinx-codeautolink>=0.15",
     "sphinx-copybutton>=0.5",
 ]
+huggingface = [
+    "huggingface-hub>0.17",
+]
 ssh = [
     "paramiko>=3.3",
     "fabric>=3",

src/experimaestro/huggingface.py

@@ -2,9 +2,16 @@
 from typing import Optional, Union
 from experimaestro.core.context import SerializationContext, SerializedPath
 from experimaestro.core.objects import ConfigInformation, ConfigMixin
-from huggingface_hub import ModelHubMixin, hf_hub_download, snapshot_download
 import os
 
+try:
+    from huggingface_hub import ModelHubMixin, hf_hub_download, snapshot_download
+except ImportError:
+    raise ImportError(
+        "huggingface-hub is required for ExperimaestroHFHub. "
+        "Install it with: pip install experimaestro[huggingface]"
+    )
+
 
 class ExperimaestroHFHub(ModelHubMixin):
     """Defines models that can be uploaded/downloaded from the Hub

src/experimaestro/launchers/slurm/base.py

@@ -833,6 +833,9 @@ class SlurmOptions:
     nodelist: Optional[str] = None
     """Request a specific list of hosts"""
 
+    ntasks_per_node: Optional[int] = None
+    """Number of tasks to run on each node"""
+
     # GPU-related
     gpus: Optional[int] = None
     """Number of GPUs"""