Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Extend TensorDictPrimer default_value options #2071

Merged
merged 21 commits into from
Apr 18, 2024
Merged

[Feature] Extend TensorDictPrimer default_value options #2071

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
c46e0fd
extend_primer_transform
albertbou92 Apr 9, 2024
822c04c
extend_primer_transform
albertbou92 Apr 9, 2024
c5f1a47
extend_primer_transform
albertbou92 Apr 9, 2024
667e21e
extend_primer_transform
albertbou92 Apr 9, 2024
e088240
extend_primer_transform
albertbou92 Apr 9, 2024
28525cd
extend_primer_transform
albertbou92 Apr 9, 2024
9e6c415
fix test
albertbou92 Apr 9, 2024
657bf9f
fix test
albertbou92 Apr 9, 2024
25fc14a
fix test
albertbou92 Apr 9, 2024
b97f727
__repr__
albertbou92 Apr 9, 2024
79f8772
mv validated
albertbou92 Apr 9, 2024
8895b62
fix tests
albertbou92 Apr 10, 2024
e3dcfb9
minor fix
albertbou92 Apr 10, 2024
a1cb9a1
minor fix
albertbou92 Apr 14, 2024
5a86c25
suggested changes
albertbou92 Apr 14, 2024
a64a97e
feedback changes
albertbou92 Apr 16, 2024
311f40b
feedback changes
albertbou92 Apr 16, 2024
3219150
feedback changes
albertbou92 Apr 16, 2024
891bd42
fix tests
albertbou92 Apr 16, 2024
9853121
Merge remote-tracking branch 'origin/main' into extend_primer_transform
vmoens Apr 18, 2024
1625514
amend
vmoens Apr 18, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
feedback changes
  • Loading branch information
@albertbou92
albertbou92 committed Apr 16, 2024
commit 311f40b026e14e4a47e2413fbcc0162ed3d9f65a
8 changes: 2 additions & 6 deletions torchrl/envs/transforms/transforms.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4521,15 +4521,11 @@ def __init__(
) # if not random and no default value, use 0.0
self.random = random
if isinstance(default_value, dict):
primer_keys = self.primers.keys(True, True)
default_value_keys = {unravel_key(key) for key in default_value.keys()}
if primer_keys != default_value_keys:
default_value = TensorDict(default_value, []).to_dict()
if set(default_value.keys()) != set(self.primers.keys(True, True)):
raise ValueError(
"If a default_value dictionary is provided, it must match the primers keys."
)
default_value = {
key: default_value[key] for key in self.primers.keys(True, True)
}
else:
default_value = {
key: default_value for key in self.primers.keys(True, True)
Expand Down