fix to ogbg CI test #5109
priyakasimbegCI.yml
on: push
fastmri
5m 6s
wmt_jax
12m 12s
wmt_pytorch
12m 53s
imagenet_jax
5m 53s
imagenet_pytorch
3m 34s
criteo_jax
3m 43s
criteo_pytorch
3m 29s
speech_jax
5m 23s
speech_pytorch
4m 21s
ogbg
3m 57s
pytest-params
7m 42s
pytest-baselines
4m 18s