temp fix for build (#5006)

ray-project · Jun 21, 2019 · 1d17125 · 1d17125
1 parent e59e807
commit 1d17125
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/ray/rllib/tests/test_optimizers.py b/python/ray/rllib/tests/test_optimizers.py
@@ -125,14 +125,14 @@ def testSimple(self):
     def testMultiGPU(self):
         local, remotes = self._make_evs()
         workers = WorkerSet._from_existing(local, remotes)
-        optimizer = AsyncSamplesOptimizer(workers, num_gpus=2, _fake_gpus=True)
+        optimizer = AsyncSamplesOptimizer(workers, num_gpus=1, _fake_gpus=True)
         self._wait_for(optimizer, 1000, 1000)
 
     def testMultiGPUParallelLoad(self):
         local, remotes = self._make_evs()
         workers = WorkerSet._from_existing(local, remotes)
         optimizer = AsyncSamplesOptimizer(
-            workers, num_gpus=2, num_data_loader_buffers=2, _fake_gpus=True)
+            workers, num_gpus=1, num_data_loader_buffers=1, _fake_gpus=True)
         self._wait_for(optimizer, 1000, 1000)
 
     def testMultiplePasses(self):
@@ -211,21 +211,21 @@ def testRejectBadConfigs(self):
                 num_data_loader_buffers=2, minibatch_buffer_size=4))
         optimizer = AsyncSamplesOptimizer(
             workers,
-            num_gpus=2,
+            num_gpus=1,
             train_batch_size=100,
             sample_batch_size=50,
             _fake_gpus=True)
         self._wait_for(optimizer, 1000, 1000)
         optimizer = AsyncSamplesOptimizer(
             workers,
-            num_gpus=2,
+            num_gpus=1,
             train_batch_size=100,
             sample_batch_size=25,
             _fake_gpus=True)
         self._wait_for(optimizer, 1000, 1000)
         optimizer = AsyncSamplesOptimizer(
             workers,
-            num_gpus=2,
+            num_gpus=1,
             train_batch_size=100,
             sample_batch_size=74,
             _fake_gpus=True)