hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[train] Fix assertion in test (#20893)

Browse source
This commit is contained in:
Siyuan (Ryans) Zhuang 2021-12-05 17:47:43 -08:00 • committed by GitHub
parent b4790900f5
commit 3285bb6b9f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/train/tests/test_trainer.py
View file

@ -298,7 +298,7 @@ def test_run_iterator(ray_start_2_cpus):
count = 0
for results in iterator:
assert (value["index"] == count for value in results)
assert all(value["index"] == count for value in results)
count += 1
assert count == 3