Fix: Allow hashing of metrics with lists in their state

Lightning-AI · Feb 12, 2021 · 056b32b · 056b32b
1 parent 4bdf2fe
commit 056b32b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pytorch_lightning/metrics/metric.py b/pytorch_lightning/metrics/metric.py
@@ -333,7 +333,11 @@ def __hash__(self):
         hash_vals = [self.__class__.__name__]
 
         for key in self._defaults.keys():
-            hash_vals.append(getattr(self, key))
+            val = getattr(self, key)
+            # Special case: allow list values, so long as their elements are hashable
+            if isinstance(val, list):
+                val = tuple(val)
+            hash_vals.append(val)
 
         return hash(tuple(hash_vals))