simplify guard

pytorch · Jul 25, 2023 · 835f40c · 835f40c
1 parent 99c6700
commit 835f40c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/test_transforms_v2_refactored.py b/test/test_transforms_v2_refactored.py
@@ -1669,8 +1669,12 @@ def forward(self, image, label):
     )
     @pytest.mark.parametrize("unpack", [True, False])
     def test_packed_unpacked(self, transform_clss, unpack):
-        if (unpack and any(issubclass(cls, self.PackedInputTransform) for cls in transform_clss)) or (
-            not unpack and any(issubclass(cls, self.UnpackedInputTransform) for cls in transform_clss)
+        if any(
+            unpack
+            and issubclass(cls, self.PackedInputTransform)
+            or not unpack
+            and issubclass(cls, self.UnpackedInputTransform)
+            for cls in transform_clss
         ):
             return