Update src/anomalib/deploy/inferencers/torch_inferencer.py

Author: samet-akcay

src/anomalib/deploy/inferencers/torch_inferencer.py

@@ -206,10 +206,7 @@ def pre_process(self, image: np.ndarray) -> torch.Tensor:
         Returns:
             Tensor: pre-processed image.
         """
-        if len (image.shape) == 3:
-            transposed_image = image.transpose(2,0,1).copy()
-        else:
-            transposed_image = image.copy()
+        transposed_image = image.transpose(2, 0, 1).copy() if len(image.shape) == 3 else image.copy()
 
         if len(transposed_image) == 3:
             return torch.from_numpy(transposed_image).float().to(self.device).unsqueeze(0)