wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

small tweaks

This commit is contained in:
Thomas Wolf 2024-05-08 16:46:15 +02:00
parent 26d9a070d8
commit f80902b171
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CONTRIBUTING.md
View File

@ -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: To develop on 🤗 LeRobot, you will at least need to install the `dev` and `test` extras dependencies along with the core library:
```bash ```bash
pip install poetry
poetry install --sync --extras "dev test" poetry install --sync --extras "dev test"
``` ```
You can also install the project with all its dependencies (including environments): You can also install the project with all its dependencies (including environments):
```bash ```bash
pip install poetry
poetry install --sync --all-extras poetry install --sync --all-extras
``` ```

6
lerobot/common/logger.py
View File

@ -84,7 +84,11 @@ class Logger:
if self._wandb and not self._disable_wandb_artifact: if self._wandb and not self._disable_wandb_artifact:
# note wandb artifact does not accept ":" in its name # note wandb artifact does not accept ":" in its name
artifact = self._wandb.Artifact( artifact = self._wandb.Artifact(
self._group.replace(":", "_") + "-" + str(self._seed) + "-" + str(identifier), self._group.replace(":", "_").replace("/", "__")
+ "-"
+ str(self._seed)
+ "-"
+ str(identifier),
type="model", type="model",
) )
artifact.add_file(save_dir / SAFETENSORS_SINGLE_FILE) artifact.add_file(save_dir / SAFETENSORS_SINGLE_FILE)