From 7e08564003f63c42d2538a16e869177803ea4d05 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Sat, 16 Mar 2024 18:21:05 +0100 Subject: [PATCH] Update torchrl dependency --- .github/poetry/cpu/poetry.lock | 16 ++++++++-------- .github/poetry/cpu/pyproject.toml | 2 +- poetry.lock | 12 ++++++------ pyproject.toml | 2 +- 4 files changed, 16 insertions(+), 16 deletions(-) diff --git a/.github/poetry/cpu/poetry.lock b/.github/poetry/cpu/poetry.lock index c07e34395..7d433c799 100644 --- a/.github/poetry/cpu/poetry.lock +++ b/.github/poetry/cpu/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 1.8.1 and should not be changed by hand. +# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand. [[package]] name = "absl-py" @@ -2903,7 +2903,7 @@ reference = "torch-cpu" [[package]] name = "torchrl" -version = "0.4.0+13bef42" +version = "0.4.0+c371266" description = "" optional = false python-versions = "*" @@ -2918,13 +2918,13 @@ tensordict = ">=0.4.0" torch = ">=2.1.0" [package.extras] -all = ["ale-py", "atari-py", "dm_control", "git", "gym", "gym[accept-rom-license]", "gymnasium", "h5py", "huggingface_hub", "hydra-core (>=1.1)", "hydra-submitit-launcher", "minari", "moviepy", "mujoco", "pandas", "pettingzoo (>=1.24.1)", "pillow", "pygame", "pytest", "pytest-instafail", "pyyaml", "requests", "scikit-learn", "scipy", "tensorboard", "torchsnapshot", "torchvision", "tqdm", "vmas (>=1.2.10)", "wandb"] +all = ["ale-py", "atari-py", "dm-control", "git", "gym", "gym[accept-rom-license]", "gymnasium", "h5py", "huggingface-hub", "hydra-core (>=1.1)", "hydra-submitit-launcher", "minari", "moviepy", "mujoco", "pandas", "pettingzoo (>=1.24.1)", "pillow", "pygame", "pytest", "pytest-instafail", "pyyaml", "requests", "scikit-learn", "scipy", "tensorboard", "torchsnapshot", "torchvision", "tqdm", "vmas (>=1.2.10)", "wandb"] atari = ["ale-py", "atari-py", "gym", "gym[accept-rom-license]", "pygame"] checkpointing = ["torchsnapshot"] -dm-control = ["dm_control"] +dm-control = ["dm-control"] gym-continuous = ["gymnasium", "mujoco"] marl = ["pettingzoo (>=1.24.1)", "vmas (>=1.2.10)"] -offline-data = ["h5py", "huggingface_hub", "minari", "pandas", "pillow", "requests", "scikit-learn", "torchvision", "tqdm"] +offline-data = ["h5py", "huggingface-hub", "minari", "pandas", "pillow", "requests", "scikit-learn", "torchvision", "tqdm"] rendering = ["moviepy"] tests = ["pytest", "pytest-instafail", "pyyaml", "scipy"] utils = ["git", "hydra-core (>=1.1)", "hydra-submitit-launcher", "tensorboard", "tqdm", "wandb"] @@ -2932,8 +2932,8 @@ utils = ["git", "hydra-core (>=1.1)", "hydra-submitit-launcher", "tensorboard", [package.source] type = "git" url = "https://github.com/pytorch/rl" -reference = "13bef426dcfa5887c6e5034a6e9697993fa92c37" -resolved_reference = "13bef426dcfa5887c6e5034a6e9697993fa92c37" +reference = "HEAD" +resolved_reference = "c371266ce5a71cb4b1a319cc56ad59d9b492cb9d" [[package]] name = "torchvision" @@ -3123,4 +3123,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p [metadata] lock-version = "2.0" python-versions = "^3.10" -content-hash = "4aa6a1e3f29560dd4a1c24d493ee1154089da4aa8d2190ad1f786c125ab2b735" +content-hash = "4dca0197360d62d56d870cd4791a9bb8bfa179645b892441a1296ac45285b19f" diff --git a/.github/poetry/cpu/pyproject.toml b/.github/poetry/cpu/pyproject.toml index fd7eb226a..6ee35a960 100644 --- a/.github/poetry/cpu/pyproject.toml +++ b/.github/poetry/cpu/pyproject.toml @@ -41,7 +41,7 @@ numba = "^0.59.0" mpmath = "^1.3.0" torch = {version = "^2.2.1", source = "torch-cpu"} tensordict = {git = "https://github.com/pytorch/tensordict"} -torchrl = {git = "https://github.com/pytorch/rl", rev = "13bef426dcfa5887c6e5034a6e9697993fa92c37"} +torchrl = {git = "https://github.com/pytorch/rl"} mujoco = "^3.1.2" mujoco-py = "^2.1.2.14" gym = "^0.26.2" diff --git a/poetry.lock b/poetry.lock index a76858bd0..134d18510 100644 --- a/poetry.lock +++ b/poetry.lock @@ -1,4 +1,4 @@ -# This file is automatically @generated by Poetry 1.8.1 and should not be changed by hand. +# This file is automatically @generated by Poetry 1.8.2 and should not be changed by hand. [[package]] name = "absl-py" @@ -2949,7 +2949,7 @@ mpmath = ">=0.19" [[package]] name = "tensordict" -version = "0.4.0+551331d" +version = "0.4.0+f1c833e" description = "" optional = false python-versions = "*" @@ -2970,7 +2970,7 @@ tests = ["pytest", "pytest-benchmark", "pytest-instafail", "pytest-rerunfailures type = "git" url = "https://github.com/pytorch/tensordict" reference = "HEAD" -resolved_reference = "ed22554d6860731610df784b2f5d09f31d3dbc7a" +resolved_reference = "f1c833ecf495aa61f3f76bf09f94dd708db496ec" [[package]] name = "termcolor" @@ -3103,8 +3103,8 @@ utils = ["git", "hydra-core (>=1.1)", "hydra-submitit-launcher", "tensorboard", [package.source] type = "git" url = "https://github.com/pytorch/rl" -reference = "13bef426dcfa5887c6e5034a6e9697993fa92c37" -resolved_reference = "13bef426dcfa5887c6e5034a6e9697993fa92c37" +reference = "HEAD" +resolved_reference = "c371266ce5a71cb4b1a319cc56ad59d9b492cb9d" [[package]] name = "torchvision" @@ -3327,4 +3327,4 @@ testing = ["big-O", "jaraco.functools", "jaraco.itertools", "more-itertools", "p [metadata] lock-version = "2.0" python-versions = "^3.10" -content-hash = "ee86b84a795e6a3e9c2d79f244a87b55589adbe46d549ac38adf48be27c04cf9" +content-hash = "e9f0d66c6b070050c1cc29b3bfbb0a1383bd356536091dbebf664c9a0db1f57c" diff --git a/pyproject.toml b/pyproject.toml index 2e818a440..7ac21d1d8 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -41,7 +41,7 @@ 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"} +torchrl = {git = "https://github.com/pytorch/rl"} mujoco = "2.3.7" mujoco-py = "^2.1.2.14" gym = "^0.26.2"