[Minor] Remove prints (#379)

pytorch · May 9, 2023 · 361edc1 · 361edc1
1 parent 961bf9d
commit 361edc1
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tensordict/nn/functional_modules.py b/tensordict/nn/functional_modules.py
@@ -200,15 +200,13 @@ def incompatible_error():
                 new_batch_size.insert(out_dim, batch_size)
                 new_names = list(batched_output.names)
                 new_names.insert(out_dim, None)
-                print(batched_output.names)
                 out = batched_output.apply(
                     lambda x, out_dim=out_dim: _remove_batch_dim(
                         x, vmap_level, batch_size, out_dim
                     ),
                     batch_size=new_batch_size,
                     names=new_names,
                 )
-                print(new_batch_size, out_dim, batch_size, flat_out_dims, out.names)
             flat_outputs.append(out)
         return tree_unflatten(flat_outputs, output_spec)