t0-0 commited on
Commit
e93d67f
·
verified ·
1 Parent(s): d834e0a

Update src/display/utils.py

Browse files
Files changed (1) hide show
  1. src/display/utils.py +4 -0
src/display/utils.py CHANGED
@@ -101,6 +101,10 @@ class Precision(Enum):
101
  return Precision.bfloat16
102
  return Precision.Unknown
103
 
 
 
 
 
104
  # Column selection
105
  COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
106
 
 
101
  return Precision.bfloat16
102
  return Precision.Unknown
103
 
104
+ class NumFewShotsType(Enum):
105
+ shot0 = ModelDetails("0-shots")
106
+ shot4 = ModelDetails("4-shots")
107
+
108
  # Column selection
109
  COLS = [c.name for c in fields(AutoEvalColumn) if not c.hidden]
110