Update src/display/utils.py

src/display/utils.py

@@ -105,6 +105,13 @@ class NumFewShotsType(Enum):
     shot0 = ModelDetails("0-shots")
     shot4 = ModelDetails("4-shots")
 
+    def from_str(num_few_shots):
+        if num_few_shots=="0":
+            return NumFewShotsType.shot0
+        if num_few_shots=="4":
+            return NumFewShotsType.shot4
+        assert True
+
 # Column selection
 COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]