Skip to content

Commit

Permalink
fix #1731 (#1732)
Browse files Browse the repository at this point in the history
Signed-off-by: YunLiu <[email protected]>
  • Loading branch information
KumoLiu authored Aug 13, 2024
1 parent 159e142 commit 364c693
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions sample-apps/endoscopy/lib/trainers/deepedit.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
import torch
from monai.apps.deepgrow.transforms import AddRandomGuidanced, FindDiscrepancyRegionsd
from monai.handlers import MeanDice, from_engine
from monai.handlers.ignite_metric import IgniteMetric
from monai.handlers.ignite_metric import IgniteMetricHandler
from monai.inferers import SimpleInferer
from monai.losses import DiceLoss
from monai.metrics import MeanIoU
Expand Down Expand Up @@ -149,7 +149,7 @@ def val_iteration_update(self, context: Context):
)


class MeanIoUMetric(IgniteMetric):
class MeanIoUMetric(IgniteMetricHandler):
def __init__(
self,
include_background: bool = True,
Expand Down

0 comments on commit 364c693

Please sign in to comment.