mirror of
https://github.com/vale981/ray
synced 2025-03-06 02:21:39 -05:00
[Tests] Fix test output (#10162)
* Trial 1. * Fix. * Revert "Fix." This reverts commit 26ad970f753d581f340857be30054d6954df8255. * Revert "Trial 1." This reverts commit 63f7aca5162bb40f2d5e28fb9647598cbde7ad41. * Another fix try. * Last trial. * Remove unnecessary comment. * Small fix. * Use better units. * Lint.
This commit is contained in:
parent
d45418936c
commit
8cedcdf2df
1 changed files with 7 additions and 1 deletions
|
@ -2,6 +2,7 @@ import re
|
|||
import subprocess
|
||||
import sys
|
||||
import pytest
|
||||
|
||||
import ray
|
||||
|
||||
from ray.test_utils import run_string_as_driver_nonblocking
|
||||
|
@ -53,7 +54,12 @@ def test_output():
|
|||
|
||||
if __name__ == "__main__":
|
||||
if len(sys.argv) > 1 and sys.argv[1] == "_ray_instance":
|
||||
ray.init(num_cpus=1)
|
||||
# Set object store memory very low so that it won't complain
|
||||
# about low shm memory in Linux environment.
|
||||
# The test failures currently complain it only has 2 GB memory,
|
||||
# so let's set it much lower than that.
|
||||
MB = 1000**2
|
||||
ray.init(num_cpus=1, object_store_memory=(100 * MB))
|
||||
ray.shutdown()
|
||||
else:
|
||||
sys.exit(pytest.main(["-v", __file__]))
|
||||
|
|
Loading…
Add table
Reference in a new issue