shigeki Ishida commited on
Commit
0ef9174
1 Parent(s): d79378a

Remove 'Unknown' and '?' from enum values

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/display/utils.py +6 -8
app.py CHANGED
@@ -586,7 +586,7 @@ with gr.Blocks() as demo_submission:
586
  )
587
  add_special_tokens = gr.Dropdown(
588
  label="AddSpecialTokens",
589
- choices=[i.value.name for i in AddSpecialTokens if i != AddSpecialTokens.Unknown],
590
  multiselect=False,
591
  value="False",
592
  )
 
586
  )
587
  add_special_tokens = gr.Dropdown(
588
  label="AddSpecialTokens",
589
+ choices=[i.value.name for i in AddSpecialTokens],
590
  multiselect=False,
591
  value="False",
592
  )
src/display/utils.py CHANGED
@@ -145,7 +145,6 @@ class Precision(Enum):
145
  class AddSpecialTokens(Enum):
146
  true = ModelDetails("True")
147
  false = ModelDetails("False")
148
- Unknown = ModelDetails("?")
149
 
150
 
151
  class NumFewShots(Enum):
@@ -163,23 +162,22 @@ class NumFewShots(Enum):
163
 
164
  class LLMJpEvalVersion(Enum):
165
  current = ModelDetails("v1.4.1")
166
- Unknown = ModelDetails("?")
167
 
168
- def from_str(version):
 
169
  if version == "1.4.1":
170
  return LLMJpEvalVersion.current
171
- else:
172
- return LLMJpEvalVersion.Unknown
173
 
174
 
175
  class VllmVersion(Enum):
176
  current = ModelDetails("v0.6.3.post1")
177
- Unknown = ModelDetails("?")
178
 
179
- def from_str(version):
 
180
  if version == "v0.6.3.post1":
181
  return VllmVersion.current
182
- return VllmVersion.Unknown
183
 
184
 
185
  # Column selection
 
145
  class AddSpecialTokens(Enum):
146
  true = ModelDetails("True")
147
  false = ModelDetails("False")
 
148
 
149
 
150
  class NumFewShots(Enum):
 
162
 
163
  class LLMJpEvalVersion(Enum):
164
  current = ModelDetails("v1.4.1")
 
165
 
166
+ @staticmethod
167
+ def from_str(version: str) -> "LLMJpEvalVersion":
168
  if version == "1.4.1":
169
  return LLMJpEvalVersion.current
170
+ raise ValueError(f"Unsupported LLMJpEval version: {version}")
 
171
 
172
 
173
  class VllmVersion(Enum):
174
  current = ModelDetails("v0.6.3.post1")
 
175
 
176
+ @staticmethod
177
+ def from_str(version: str) -> "VllmVersion":
178
  if version == "v0.6.3.post1":
179
  return VllmVersion.current
180
+ raise ValueError(f"Unsupported VLLM version: {version}")
181
 
182
 
183
  # Column selection