[RLlib] Fix two test cases that only fail on Travis. (#11435)

2026-09-09 11:32:43 +08:00 · 2020-10-16 13:53:30 -05:00
parent f890808c14
commit 2aec77e305
4 changed files with 12 additions and 5 deletions
@@ -10,7 +10,7 @@ class TestDistributedExecution(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
-        ray.init(ignore_reinit_error=True)
+        ray.init(num_cpus=4)

    @classmethod
    def tearDownClass(cls):