brunneis commited on
Commit
aaed00b
Β·
unverified Β·
1 Parent(s): 7592671

Fix BrainDAO model type

Browse files
src/display/utils.py CHANGED
@@ -111,7 +111,7 @@ class ModelType(Enum):
111
  if "pretrained" in type or "πŸ’Ž" in type:
112
  return ModelType.PT
113
  if "braindao" in type or "🧠" in type:
114
- return ModelType.IFT
115
  return ModelType.Unknown
116
 
117
 
 
111
  if "pretrained" in type or "πŸ’Ž" in type:
112
  return ModelType.PT
113
  if "braindao" in type or "🧠" in type:
114
+ return ModelType.BrainDAO
115
  return ModelType.Unknown
116
 
117
 
src/leaderboard/read_evals.py CHANGED
@@ -108,8 +108,9 @@ class EvalResult:
108
  self.likes = request.get("likes", 0)
109
  self.num_params = request.get("params", 0)
110
  self.date = request.get("submitted_time", "")
111
- except Exception:
112
  print(f"Could not find request file for {self.org}/{self.model} with precision {self.precision.value.name}")
 
113
 
114
  def to_dict(self):
115
  """Converts the Eval Result to a dict compatible with our dataframe display"""
 
108
  self.likes = request.get("likes", 0)
109
  self.num_params = request.get("params", 0)
110
  self.date = request.get("submitted_time", "")
111
+ except Exception as error:
112
  print(f"Could not find request file for {self.org}/{self.model} with precision {self.precision.value.name}")
113
+ print(f"Error: {error}")
114
 
115
  def to_dict(self):
116
  """Converts the Eval Result to a dict compatible with our dataframe display"""