amend

pytorch · Jan 11, 2024 · 5d3787f · 5d3787f
1 parent ccc06ca
commit 5d3787f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -31,7 +31,7 @@ repos:
           - flake8-print==5.0.0
 
   - repo: https://github.com/PyCQA/pydocstyle
-    rev: 6.3.0
+    rev: 6.1.1
     hooks:
       - id: pydocstyle
         files: ^tensordict/
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -3091,7 +3091,7 @@ def _recv(
             if isinstance(value, Tensor):
                 pass
             elif _is_tensor_collection(value.__class__):
-                _tag = value._recv(src, _tag=_tag, pseudo_rand=pseudo_rand)
+                _tag = value._recv(src, _tag=_tag, pseudo_rand=pseudo_rand, group=group)
                 continue
             else:
                 raise NotImplementedError(f"Type {type(value)} is not supported.")