fix tests

Michedev · Nov 15, 2022 · 34dcdc8 · 34dcdc8
1 parent 33242aa
commit 34dcdc8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/workflows/python-publish.yml b/.github/workflows/python-publish.yml
@@ -36,7 +36,10 @@ jobs:
         pip install build
         python -m pip install .[dev]
     - name: Test model/ folder
-      run: python -m pytest ddpm
+      run: |
+           mv ddpm model
+           python -m pytest
+           mv model ddpm
     - name: Build package
       run: python -m build
     - name: Publish package

diff --git a/model/test_unet.py → tests/test_unet.py b/model/test_unet.py → tests/test_unet.py
@@ -1,5 +1,5 @@
 import torch
-from . import UNetTimeStep
+from model import UNetTimeStep
 
 
 def test_unet_same_width_height():