We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e999f88 commit 2d5ccf3Copy full SHA for 2d5ccf3
1 file changed
rectools/metrics/scoring.py
@@ -211,4 +211,4 @@ def calc_metrics( # noqa # pylint: disable=too-many-branches,too-many-locals,t
211
if len(results) < expected_results_len:
212
warnings.warn("Custom metrics are not supported.")
213
214
- return results
+ return {k: v.item() if hasattr(v, "item") else v for k, v in results.items()}
0 commit comments