Fix BrainDAO model type
Browse files- src/display/utils.py +1 -1
- src/leaderboard/read_evals.py +2 -1
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.
|
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"""
|