From fc6a57dbc77e3f2a25fbe3a7a9fcc920cef3a97a Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Tue, 23 Jul 2024 13:24:11 +0100 Subject: [PATCH] Update [ghstack-poisoned] --- test/test_cost.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/test_cost.py b/test/test_cost.py index c3912759453..8fea54db2a5 100644 --- a/test/test_cost.py +++ b/test/test_cost.py @@ -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