Skip to content

Commit 3f64cc8

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent ee57c64 commit 3f64cc8

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

lambench/metrics/vishelper/metrics_calculations.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,9 @@ def calculate_generalizability_downstream_score(self) -> tuple[dict, dict]:
159159
) # drop models with missing domain results
160160

161161
# # Now aggregate all domains to get the final generalizability metrics for each model
162-
return domain_results.to_dict(orient="index"), domain_results.mean(axis=1).to_dict()
162+
return domain_results.to_dict(orient="index"), domain_results.mean(
163+
axis=1
164+
).to_dict()
163165

164166
def calculate_stability_results(self) -> dict[str, float]:
165167
"""This calculates the stability score for a given LAM."""

0 commit comments

Comments
 (0)