From f80902b171c0c57c420be849bef2c99a99c6f4a4 Mon Sep 17 00:00:00 2001 From: Thomas Wolf <thomas@huggingface.co> Date: Wed, 8 May 2024 16:46:15 +0200 Subject: [PATCH] small tweaks --- CONTRIBUTING.md | 2 ++ lerobot/common/logger.py | 6 +++++- 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index dd8f97e2..afa920cc 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -139,11 +139,13 @@ Follow these steps to start contributing: To develop on 🤗 LeRobot, you will at least need to install the `dev` and `test` extras dependencies along with the core library: ```bash + pip install poetry poetry install --sync --extras "dev test" ``` You can also install the project with all its dependencies (including environments): ```bash + pip install poetry poetry install --sync --all-extras ``` diff --git a/lerobot/common/logger.py b/lerobot/common/logger.py index 8e7fe7f2..dcdd545e 100644 --- a/lerobot/common/logger.py +++ b/lerobot/common/logger.py @@ -84,7 +84,11 @@ class Logger: if self._wandb and not self._disable_wandb_artifact: # note wandb artifact does not accept ":" in its name artifact = self._wandb.Artifact( - self._group.replace(":", "_") + "-" + str(self._seed) + "-" + str(identifier), + self._group.replace(":", "_").replace("/", "__") + + "-" + + str(self._seed) + + "-" + + str(identifier), type="model", ) artifact.add_file(save_dir / SAFETENSORS_SINGLE_FILE)