Update classlabel test

Author: mariosasko

src/datasets/features/features.py

@@ -830,8 +830,9 @@ def _strvalue2int(self, value: str):
                     int_value = int(value)
                 except ValueError:
                     failed_parse = True
-                if int_value < -1 or int_value >= self.num_classes:
-                    failed_parse = True
+                else:
+                    if int_value < -1 or int_value >= self.num_classes:
+                        failed_parse = True
         if failed_parse:
             raise ValueError(f"Invalid string class label {value}")
         return int_value

tests/features/test_features.py

@@ -253,7 +253,7 @@ def test_classlabel_str2int():
     classlabel = ClassLabel(names=names)
     for label in names:
         assert classlabel.str2int(label) == names.index(label)
-    with pytest.raises(KeyError):
+    with pytest.raises(ValueError):
         classlabel.str2int("__bad_label_name__")
     with pytest.raises(ValueError):
         classlabel.str2int(1)