switch to correct ignore attribute

Lightning-AI · Apr 15, 2021 · 16a642e · 16a642e
1 parent 124116e
commit 16a642e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmetrics/metric.py b/torchmetrics/metric.py
@@ -59,7 +59,7 @@ class Metric(nn.Module, ABC):
             will be used to perform the allgather.
     """
 
-    __jit_unused_properties__ = ["is_differentiable"]
+    __jit_ignored_attributes__ = ["is_differentiable"]
 
     def __init__(
         self,