elif

Author: Borda

references/classification/transforms.py

@@ -40,11 +40,11 @@ def forward(self, batch: Tensor, target: Tensor) -> Tuple[Tensor, Tensor]:
         """
         if batch.ndim != 4:
             raise ValueError(f"Batch ndim should be 4. Got {batch.ndim}")
-        elif target.ndim != 1:
+        if target.ndim != 1:
             raise ValueError(f"Target ndim should be 1. Got {target.ndim}")
-        elif not batch.is_floating_point():
+        if not batch.is_floating_point():
             raise TypeError(f"Batch dtype should be a float tensor. Got {batch.dtype}.")
-        elif target.dtype != torch.int64:
+        if target.dtype != torch.int64:
             raise TypeError(f"Target dtype should be torch.int64. Got {target.dtype}")
 
         if not self.inplace:
@@ -116,11 +116,11 @@ def forward(self, batch: Tensor, target: Tensor) -> Tuple[Tensor, Tensor]:
         """
         if batch.ndim != 4:
             raise ValueError(f"Batch ndim should be 4. Got {batch.ndim}")
-        elif target.ndim != 1:
+        if target.ndim != 1:
             raise ValueError(f"Target ndim should be 1. Got {target.ndim}")
-        elif not batch.is_floating_point():
+        if not batch.is_floating_point():
             raise TypeError(f"Batch dtype should be a float tensor. Got {batch.dtype}.")
-        elif target.dtype != torch.int64:
+        if target.dtype != torch.int64:
             raise TypeError(f"Target dtype should be torch.int64. Got {target.dtype}")
 
         if not self.inplace: