how annoying

Parry-Parry · Nov 6, 2024 · d902ed1 · d902ed1
1 parent ac1b702
commit d902ed1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rankers/train/training_arguments.py b/rankers/train/training_arguments.py
@@ -60,7 +60,7 @@ def to_dict(self):
                 d[k] = v.value
             if isinstance(v, list) and len(v) > 0 and isinstance(v[0], Enum):
                 d[k] = [x.value for x in v]
-            if isinstance(v, list) and is_ir_measures_available():
+            if isinstance(v, list) and is_ir_measures_available() and len(v) > 0:
                 import ir_measures
                 if type(v[0]) == ir_measures.Measure:
                     d[k] = [x.NAME for x in v]