amend

pytorch · Jul 11, 2024 · b0cc6f5 · b0cc6f5
1 parent 57d8cd4
commit b0cc6f5
Showing 1 changed file with 12 additions and 12 deletions.
diff --git a/torchrl/data/tensor_specs.py b/torchrl/data/tensor_specs.py
@@ -1370,9 +1370,9 @@ def n(self):
         return self.space.n
 
     def update_mask(self, mask):
-        """
-        Sets a mask to prevent some of the possible outcomes when a sample is
-        taken. The mask can also be set during initialization of the spec.
+        """Sets a mask to prevent some of the possible outcomes when a sample is taken.
+
+        The mask can also be set during initialization of the spec.
 
         Args:
             mask (torch.Tensor or None): boolean mask. If None, the mask is
@@ -2587,9 +2587,9 @@ def __init__(
         self.update_mask(mask)
 
     def update_mask(self, mask):
-        """
-        Sets a mask to prevent some of the possible outcomes when a sample is
-        taken. The mask can also be set during initialization of the spec.
+        """Sets a mask to prevent some of the possible outcomes when a sample is taken.
+
+        The mask can also be set during initialization of the spec.
 
         Args:
             mask (torch.Tensor or None): boolean mask. If None, the mask is
@@ -2980,9 +2980,9 @@ def n(self):
         return self.space.n
 
     def update_mask(self, mask):
-        """
-        Sets a mask to prevent some of the possible outcomes when a sample is
-        taken. The mask can also be set during initialization of the spec.
+        """Sets a mask to prevent some of the possible outcomes when a sample is taken.
+
+        The mask can also be set during initialization of the spec.
 
         Args:
             mask (torch.Tensor or None): boolean mask. If None, the mask is
@@ -3429,9 +3429,9 @@ def __init__(
         self.remove_singleton = remove_singleton
 
     def update_mask(self, mask):
-        """
-        Sets a mask to prevent some of the possible outcomes when a sample is
-        taken. The mask can also be set during initialization of the spec.
+        """Sets a mask to prevent some of the possible outcomes when a sample is taken.
+
+        The mask can also be set during initialization of the spec.
 
         Args:
             mask (torch.Tensor or None): boolean mask. If None, the mask is