diff --git a/sample-apps/endoscopy/lib/trainers/deepedit.py b/sample-apps/endoscopy/lib/trainers/deepedit.py index b8f018b29..ebfb954b6 100644 --- a/sample-apps/endoscopy/lib/trainers/deepedit.py +++ b/sample-apps/endoscopy/lib/trainers/deepedit.py @@ -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 @@ -149,7 +149,7 @@ def val_iteration_update(self, context: Context): ) -class MeanIoUMetric(IgniteMetric): +class MeanIoUMetric(IgniteMetricHandler): def __init__( self, include_background: bool = True,