Fixed a bug in calculating the metric for Binary labels

Author: srikesh-07

src/anomalib/models/components/base/export_mixin.py

@@ -393,6 +393,7 @@ def val_fn(nncf_model: "CompiledModel", validation_data: Iterable) -> float:
                     setattr(batch, name, torch.from_numpy(pred))
                 if batch.gt_mask is not None:
                     batch.gt_mask = batch.gt_mask.unsqueeze(dim=1)
+                batch.pred_score = batch.pred_score.squeeze(dim=1)  # Squeezing since it is binary. (B, 1) -> (B)
                 metric.update(batch)
             return metric.compute()