tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

LerobotDataset pushable to HF from any folder (#563)

Browse Source
This commit is contained in:
Claudio Coppola
2024-12-09 10:32:25 +00:00
committed by GitHub
parent 03f49ceaf0
commit 44f9b21e74
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/datasets/utils.py
View File

@@ -13,6 +13,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
import importlib.resources
import json import json
import logging import logging
import textwrap import textwrap
@@ -476,6 +477,8 @@ def create_lerobot_dataset_card(
Note: If specified, license must be one of https://huggingface.co/docs/hub/repositories-licenses. Note: If specified, license must be one of https://huggingface.co/docs/hub/repositories-licenses.
""" """
card_tags = ["LeRobot"] card_tags = ["LeRobot"]
card_template_path = importlib.resources.path("lerobot.common.datasets", "card_template.md")
if tags: if tags:
card_tags += tags card_tags += tags
if dataset_info: if dataset_info:
@@ -493,8 +496,9 @@ def create_lerobot_dataset_card(
} }
], ],
) )
return DatasetCard.from_template( return DatasetCard.from_template(
card_data=card_data, card_data=card_data,
template_path="./lerobot/common/datasets/card_template.md", template_path=str(card_template_path),
**kwargs, **kwargs,
) )