Use MONAI deprecated decorator

Signed-off-by: Soumya Snigdha Kundu <[email protected]>

Author: aymuos15

monai/apps/deepedit/transforms.py

@@ -24,7 +24,7 @@
 from monai.data import MetaTensor
 from monai.networks.layers import GaussianFilter
 from monai.transforms.transform import MapTransform, Randomizable, Transform
-from monai.utils import min_version, optional_import
+from monai.utils import deprecated, min_version, optional_import
 
 measure, _ = optional_import("skimage.measure", "0.14.2", min_version)
 
@@ -155,26 +155,18 @@ def __call__(self, data: Mapping[Hashable, np.ndarray]) -> dict[Hashable, np.nda
         return d
 
 
+@deprecated(since="1.6", removed="1.8", msg_suffix="Use `RemapLabelsToSequentiald` instead.")
 class NormalizeLabelsInDatasetd(RemapLabelsToSequentiald):
     """
-    .. deprecated:: 1.5.0
-        `NormalizeLabelsInDatasetd` is deprecated. Use :class:`RemapLabelsToSequentiald` instead.
+    .. deprecated:: 1.6.0
+        `NormalizeLabelsInDatasetd` is deprecated and will be removed in version 1.8.0.
+        Use :class:`RemapLabelsToSequentiald` instead.
 
     This class is maintained for backward compatibility. Please use RemapLabelsToSequentiald
     which better describes the transform's functionality.
     """
 
-    def __init__(
-        self, keys: KeysCollection, label_names: dict[str, int] | None = None, allow_missing_keys: bool = False
-    ):
-        warnings.warn(
-            "NormalizeLabelsInDatasetd is deprecated and will be removed in a future version. "
-            "Please use RemapLabelsToSequentiald instead, which better describes what the transform does: "
-            "remapping label values to sequential indices (0, 1, 2, 3, ...).",
-            DeprecationWarning,
-            stacklevel=2,
-        )
-        super().__init__(keys, label_names, allow_missing_keys)
+    pass
 
 
 class SingleLabelSelectiond(MapTransform):

tests/apps/deepedit/test_deepedit_transforms.py

@@ -332,19 +332,26 @@ def test_multiple_labels(self):
         self.assertEqual(result["label_names"], {"background": 0, "kidney": 1, "liver": 2, "spleen": 3})
 
     def test_deprecated_name_warning(self):
-        """Test that using the deprecated name raises a warning"""
+        """Test that using the deprecated name raises a warning when version >= 1.6"""
         import warnings
 
+        from monai.utils import deprecated
+
+        # Create a test class with version_val to simulate version 1.6
+        @deprecated(since="1.6", removed="1.8", msg_suffix="Use `RemapLabelsToSequentiald` instead.", version_val="1.6")
+        class TestDeprecatedClass(RemapLabelsToSequentiald):
+            pass
+
         data = {"label": np.array([[[0, 1]]])}
 
         with warnings.catch_warnings(record=True) as w:
             warnings.simplefilter("always")
-            transform = NormalizeLabelsInDatasetd(keys="label", label_names={"spleen": 1, "background": 0})
-            _ = transform(data)  # Call to trigger the warning
+            transform = TestDeprecatedClass(keys="label", label_names={"spleen": 1, "background": 0})
+            _ = transform(data)  # Invoke the transform to confirm functionality
 
-            # Check that a deprecation warning was raised
+            # Check that a deprecation warning was raised (deprecated decorator uses FutureWarning)
             self.assertEqual(len(w), 1)
-            self.assertTrue(issubclass(w[0].category, DeprecationWarning))
+            self.assertTrue(issubclass(w[0].category, FutureWarning))
             self.assertIn("RemapLabelsToSequentiald", str(w[0].message))