Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-06-01 21:56:38 +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
ogb_lsc/pcq/ensemble_predictions.py
+2 -1
View File
@@ -125,7 +125,8 @@ def _generate_test_prediction_file(test_predictions: np.ndarray,
# It is not possible to write directlt on CNS. # It is not possible to write directlt on CNS.
evaluator = lsc.PCQM4MEvaluator() 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 return output_path