From f138e5948a076bff69b2188d700c9c5d5a415e30 Mon Sep 17 00:00:00 2001
From: Jash Shah <49280550+jashshah999@users.noreply.github.com>
Date: Wed, 25 Feb 2026 03:29:10 -0800
Subject: [PATCH] Fix metaworld_config.json not bundled in pip installs and
 AttributeError crash (#3017)

1. Include metaworld_config.json in package distributions by adding it to
   both MANIFEST.in (for sdist) and pyproject.toml package-data (for wheels).
   Without this, pip-installed lerobot raises FileNotFoundError when
   importing the metaworld environment.

2. Fix crash in sanity_check_dataset_name where the error message accesses
   policy_cfg.type when policy_cfg is None, raising AttributeError instead
   of the intended ValueError.

Fixes #2958
---
 MANIFEST.in                        | 1 +
 pyproject.toml                     | 3 +++
 src/lerobot/utils/control_utils.py | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/MANIFEST.in b/MANIFEST.in
index c1fb2ea75..c1fce3b5a 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,2 +1,3 @@
 include src/lerobot/templates/lerobot_modelcard_template.md
 include src/lerobot/datasets/card_template.md
+include src/lerobot/envs/metaworld_config.json
diff --git a/pyproject.toml b/pyproject.toml
index ea3df4a6d..b6d85b0f6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -214,6 +214,9 @@ lerobot-edit-dataset="lerobot.scripts.lerobot_edit_dataset:main"
 lerobot-setup-can="lerobot.scripts.lerobot_setup_can:main"
 
 # ---------------- Tool Configurations ----------------
+[tool.setuptools.package-data]
+lerobot = ["envs/*.json"]
+
 [tool.setuptools.packages.find]
 where = ["src"]
 
diff --git a/src/lerobot/utils/control_utils.py b/src/lerobot/utils/control_utils.py
index 7cfe177ef..7c605af17 100644
--- a/src/lerobot/utils/control_utils.py
+++ b/src/lerobot/utils/control_utils.py
@@ -189,7 +189,7 @@ def sanity_check_dataset_name(repo_id, policy_cfg):
     # Check if dataset_name starts with "eval_" but policy is missing
     if dataset_name.startswith("eval_") and policy_cfg is None:
         raise ValueError(
-            f"Your dataset name begins with 'eval_' ({dataset_name}), but no policy is provided ({policy_cfg.type})."
+            f"Your dataset name begins with 'eval_' ({dataset_name}), but no policy is provided."
         )
 
     # Check if dataset_name does not start with "eval_" but policy is provided