Spaces:
Runtime error
Runtime error
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"""
|