Skip to content

Conversation

@tcbegley
Copy link
Contributor

Description

pytorch/tensordict#109 fully adopts the prototype probabilistic modules. Here we update the imports in TorchRL accordingly.

Should be merged alongside the tensordict changes.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Dec 12, 2022
@tcbegley tcbegley changed the title Drop prototype imports [Refactor] Drop prototype imports Dec 16, 2022
Copy link
Collaborator

@vmoens vmoens left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! Let's wait for the tests to complete and merge

@codecov
Copy link

codecov bot commented Dec 16, 2022

Codecov Report

Merging #738 (11c504d) into main (2b18311) will decrease coverage by 0.01%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main     #738      +/-   ##
==========================================
- Coverage   88.76%   88.74%   -0.02%     
==========================================
  Files         123      123              
  Lines       20810    20803       -7     
==========================================
- Hits        18471    18461      -10     
- Misses       2339     2342       +3     
Flag Coverage Δ
habitat-gpu 25.17% <100.00%> (-0.01%) ⬇️
linux-brax 29.56% <100.00%> (-0.01%) ⬇️
linux-cpu 85.26% <100.00%> (-0.01%) ⬇️
linux-gpu 86.18% <100.00%> (-0.02%) ⬇️
linux-jumanji 30.30% <100.00%> (-0.01%) ⬇️
linux-outdeps-gpu 72.01% <100.00%> (-0.01%) ⬇️
linux-stable-cpu 85.11% <100.00%> (-0.01%) ⬇️
linux-stable-gpu 85.82% <100.00%> (-0.01%) ⬇️
linux_examples-gpu 43.13% <100.00%> (-0.01%) ⬇️
macos-cpu 84.94% <100.00%> (-0.01%) ⬇️
olddeps-gpu 75.91% <100.00%> (-0.01%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
test/test_tensordictmodules.py 96.86% <ø> (-0.03%) ⬇️
torchrl/modules/tensordict_module/probabilistic.py 78.84% <100.00%> (-0.40%) ⬇️
torchrl/envs/vec_env.py 68.61% <0.00%> (-0.51%) ⬇️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@vmoens vmoens merged commit 1f8341c into pytorch:main Dec 16, 2022
@tcbegley tcbegley deleted the drop-prototype branch December 16, 2022 15:02
vmoens added a commit that referenced this pull request Dec 23, 2022
* [BugFix] Fix NoopReset in parallel settings (#747)

* [Refactor] Remove env.is_done attribute (#748)

* Drop use of prototype modules (#738)

* Fix test_cost

* Lint and format

* Fix test_postprocs

* Fix test_collectors

* amend

* [BugFix] Fixes for `speed` branch merge on tensordict (#755)

* init

* empty

* empty

* amend

* empty

* run legit tests

* empty

* reset_workers

* brax/jumanji fix

* lint

* reward shape for jumanji

* jumanji

* jumanji

* bf

* amend

* amend

* amend

* bf

* bf

* bf

Co-authored-by: Tom Begley <tomcbegley@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants