Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-02-06 03:32:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

pytype fix

Browse Source 
PiperOrigin-RevId: 525432152
This commit is contained in:
Alvaro Sanchez-Gonzalez
2023-04-19 14:37:09 +01:00
committed by Saran Tunyasuvunakool
parent f905943c13
commit fb1d757863
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ogb_lsc/pcq/ensemble_predictions.py
View File

@@ -125,7 +125,8 @@ def _generate_test_prediction_file(test_predictions: np.ndarray,
# It is not possible to write directlt on CNS.
evaluator = lsc.PCQM4MEvaluator()
evaluator.save_test_submission(dict(y_pred=test_predictions), output_path)
evaluator.save_test_submission(
dict(y_pred=test_predictions), str(output_path))
return output_path