no env in test

awarebayes · Oct 28, 2019 · 6b3e4ed · 6b3e4ed
1 parent c795ba1
commit 6b3e4ed
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -6,7 +6,7 @@ version: 2
 jobs:
   build:
     docker:
-      - image: pytorch/pytorch:1.3-cuda10.1-cudnn7-runtime
+      - image: pytorch/pytorch
 
 
     working_directory: ~/repo
@@ -38,7 +38,7 @@ jobs:
           command: |
             . venv/bin/activate
             pip3 install torch==1.3.0
-            python -m pip install -U matplotlib
+            pip3 install matplotlib
             pytest -v -s ./.circleci/tests/main.py
 
       - store_artifacts:

diff --git a/.circleci/tests/learning.py b/.circleci/tests/learning.py
@@ -80,8 +80,7 @@ def test_algo():
     value_net  = recnn.nn.Critic(1290, 128, 256, 54e-2)
     policy_net = recnn.nn.Actor(1290, 128, 256, 6e-1)
 
-    cuda = torch.device('cuda')
     ddpg = recnn.nn.DDPG(policy_net, value_net)
-    ddpg = ddpg.to(cuda)
+    ddpg = ddpg
     loss = ddpg.update(batch, learn=True)
     check_loss_and_networks(loss, ddpg.nets)