Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
vmoens committed Jul 23, 2024
1 parent a4cdd5e commit fc6a57d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions test/test_cost.py
Original file line number Diff line number Diff line change
Expand Up @@ -14674,8 +14674,10 @@ def test_from_module_list(self):
class MyLoss(LossModule):
module_a: TensorDictModule
module_b: TensorDictModule

module_a_params: TensorDict
module_b_params: TensorDict

target_module_a_params: TensorDict
target_module_b_params: TensorDict

Expand Down

0 comments on commit fc6a57d

Please sign in to comment.