Added poetry setup

.gitignore

@@ -1,3 +1,6 @@
+# Custom
+diffusion_policy
+
 # Logging
 logs
 tmp

README.md

@@ -2,52 +2,32 @@
 
 ## Installation
 
-Install dependencies using `conda`:
+Create a virtual environment with python 3.10, e.g. using `conda`:
-
 ```
-conda env create -f environment.yaml
+conda create -y -n lerobot python=3.10
 conda activate lerobot
 ```
 
-Install `torchrl`, `tensordict` and `diffusion_policy` dev builds
+[Install `poetry`](https://python-poetry.org/docs/#installation) (if you don't have it already)
+```
+curl -sSL https://install.python-poetry.org | python3 -
+```
+
+Install dependencies
+```
+poetry install
+```
+
+If you encounter a disk space error, try to change your tmp dir to a location where you have enough disk space, e.g.
+```
+mkdir ~/tmp
+export TMPDIR='~/tmp'
+```
+
+Install `diffusion_policy` #HACK
 ```
-cd path/to/root
-git clone https://github.com/pytorch/tensordict
-git clone https://github.com/pytorch/rl
 git clone https://github.com/real-stanford/diffusion_policy
-cd tensordict
+cp -r diffusion_policy/diffusion_policy $(poetry env info -p)/lib/python3.10/site-packages/
-python setup.py develop
-cd ../rl
-python setup.py develop
-cd ../diffusion_policy
-python setup.py develop
-```
-
-Install additional modules
-```
-pip install \
-    hydra \
-    termcolor \
-    einops \
-    pygame \
-    pymunk \
-    zarr \
-    gym \
-    shapely \
-    opencv-python \
-    scikit-image \
-    mpmath==1.3.0 \
-```
-
-Fix Hydra
-```
-pip install hydra-core --upgrade
-```
-
-**dev**
-
-```
-python setup.py develop
 ```
 
 ## Usage

pyproject.toml

@@ -0,0 +1,55 @@
+[tool.poetry]
+name = "lerobot"
+version = "0.1.0"
+description = "Le robot is learning"
+authors = [
+    "Rémi Cadène <re.cadene@gmail.com>",
+    "Simon Alibert <alibert.sim@gmail.com>",
+]
+repository = "https://github.com/Cadene/lerobot"
+readme = "README.md"
+license = "MIT"
+classifiers=[
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Topic :: Software Development :: Build Tools",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3.8",
+]
+packages = [{include = "lerobot"}]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+cython = "^3.0.8"
+termcolor = "^2.4.0"
+omegaconf = "^2.3.0"
+dm-env = "^1.6"
+pandas = "^2.2.1"
+wandb = "^0.16.3"
+moviepy = "^1.0.3"
+imageio = {extras = ["pyav"], version = "^2.34.0"}
+gdown = "^5.1.0"
+hydra-core = "^1.3.2"
+einops = "^0.7.0"
+pygame = "^2.5.2"
+pymunk = "^6.6.0"
+zarr = "^2.17.0"
+shapely = "^2.0.3"
+scikit-image = "^0.22.0"
+numba = "^0.59.0"
+mpmath = "^1.3.0"
+torch = "^2.2.1"
+tensordict = {git = "https://github.com/pytorch/tensordict"}
+torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"}
+mujoco = "^3.1.2"
+mujoco-py = "^2.1.2.14"
+gym = "^0.26.2"
+opencv-python = "^4.9.0.80"
+diffusion-policy = {git = "https://github.com/real-stanford/diffusion_policy"}
+
+
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"