Update on "Update pipeline API to accept arbitrary sequence of Tensor…

…s and not just Tuple"

The current API's forward method only accepted a Tensor or a Tuple of
Tensors, making this more generic by accepting any Sequence of Tensors.

Differential Revision: [D25181944](https://our.internmc.facebook.com/intern/diff/D25181944/)

[ghstack-poisoned]

torch/distributed/_pipeline/sync/microbatch.py

@@ -110,7 +110,7 @@ def __setitem__(self, index: Union[int, slice], value: TensorOrTensors) -> None:
     def _setitem_by_index(self, index: int, value: Tensor) -> None:
         if not self.atomic:
             i = index
-            self.value = self.value[:i] + (value,) + self.value[i + 1 :] #type: ignore
+            self.value = self.value[:i] + (value,) + self.value[i + 1 :]  # type: ignore
             return
 
         if index != 0: