remove fit comments. small typo of shape dimensions

Signed-off-by: Alfie Roddan <228966941+alfieroddanintel@users.noreply.github.com>

Author: alfieroddanintel

src/anomalib/models/image/anomaly_dino/lightning_model.py

@@ -187,13 +187,13 @@ def __init__(
     @classmethod
     def configure_pre_processor(
         cls,
-        image_size: tuple[int, int] | None = None,
+        image_size: tuple[int, int] | int | None = None,
     ) -> PreProcessor:
         """Configure the default pre-processor for AnomalyDINO.
 
         Args:
-            image_size (tuple[int, int] | None, optional): Target size for resizing
-                input images. Defaults to ``(252, 252)``.
+            image_size (tuple[int, int] | int | None, optional): Target size for resizing
+                input images. Defaults to ``(252, 252)``. Note if int, keeps aspect ratio and resizes shortest side.
 
         Returns:
             PreProcessor: Configured pre-processor instance.
@@ -243,11 +243,8 @@ def fit(self) -> None:
         """Optional fitting step.
 
         This method is a placeholder for potential post-training operations
-        such as coreset subsampling or feature normalization.
-
-        Note:
-            The current implementation is a no-op, as AnomalyDINO typically
-            performs inference directly after feature extraction.
+        such as coreset subsampling or feature normalization. The model
+        handles fitting (if-needed).
         """
         self.model.fit()
 

src/anomalib/models/image/anomaly_dino/torch_model.py

@@ -224,7 +224,7 @@ def forward(self, input_tensor: torch.Tensor) -> torch.Tensor | InferenceBatch:
         input_tensor = input_tensor.type(self.memory_bank.dtype)
 
         # work out sizing
-        b, _, w, h = input_tensor.shape
+        b, _, h, w = input_tensor.shape
         cropped_width = w - w % self.feature_encoder.patch_size
         cropped_height = h - h % self.feature_encoder.patch_size
         grid_size = (