Merge pull request #631 from mv1388/pytorch-1-7-fix

Fix moto version
mv1388 · Oct 30, 2020 · fa7b319 · fa7b319
2 parents 165aa0d + 8823f1b
commit fa7b319
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/.github/workflows/package_testing.yml b/.github/workflows/package_testing.yml
@@ -23,7 +23,8 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         pip install 'six>=1.13.0'
-        pip install tensorflow keras moto[all] pytest
+        pip install 'moto[all]==1.3.14' pytest
+        pip install tensorflow keras
         pip install -r requirements.txt
 
     - name: Lint with flake8

diff --git a/.travis.yml b/.travis.yml
@@ -40,7 +40,8 @@ matrix:
 # command to install dependencies
 install:
   - pip install 'six>=1.13.0'
-  - pip install tensorflow keras moto[all] pytest
+  - pip install 'moto[all]==1.3.14' pytest
+  - pip install tensorflow keras
   - pip install -r requirements.txt
 
 script:

diff --git a/tests/utils.py b/tests/utils.py
@@ -85,7 +85,7 @@ def forward(self, x):
     def get_loss(self, batch_data, criterion, device):
         x, y = batch_data
         pred_y = self(x)
-        loss = criterion(pred_y, y)
+        loss = criterion(pred_y.squeeze(), y)
         return loss
 
     def get_predictions(self, batch_data, device):