wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-13 12:30:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[docker] Detect CPUs in container correctly (#10507)

Browse Source 
Co-authored-by: simon-mo <simon.mo@hey.com>
Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
Co-authored-by: Alex Wu <itswu.alex@gmail.com>
This commit is contained in:
Ian Rodney
2020-09-14 17:11:47 +00:00
committed by Barak Michener
co-authored by simon-mo Richard Liaw Alex Wu
parent fe23f23680
commit 826a9253c6
35 changed files with 173 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_ray_trial_executor.py
+1 -1
View File
@@ -16,7 +16,7 @@ from ray.cluster_utils import Cluster
class RayTrialExecutorTest(unittest.TestCase):
def setUp(self):
self.trial_executor = RayTrialExecutor(queue_trials=False)
ray.init(ignore_reinit_error=True)
ray.init(num_cpus=2, ignore_reinit_error=True)
_register_all() # Needed for flaky tests
def tearDown(self):