Revert "fix dice score import"

This reverts commit 71f0539.

Author: ziw-liu

viscy/translation/evaluation.py

@@ -3,8 +3,7 @@
 import logging
 
 from lightning.pytorch import LightningModule
-from torchmetrics.functional import accuracy, jaccard_index
-from torchmetrics.functional.segmentation import dice_score
+from torchmetrics.functional import accuracy, dice, jaccard_index
 
 from viscy.data.typing import SegmentationSample
 from viscy.translation.evaluation_metrics import mean_average_precision
@@ -37,7 +36,7 @@ def test_step(self, batch: SegmentationSample, batch_idx: int) -> None:
                 "position": batch["position_idx"][0],
                 "time": batch["time_idx"][0],
                 "accuracy": (accuracy(pred_binary, target_binary, task="binary")),
-                "dice": (dice_score(pred_binary, target_binary)),
+                "dice": (dice(pred_binary, target_binary)),
                 "jaccard": (jaccard_index(pred_binary, target_binary, task="binary")),
                 "mAP": coco_metrics["map"],
                 "mAP_50": coco_metrics["map_50"],