Update build.yml

masadcv · Oct 18, 2023 · 9a8b6e9 · 9a8b6e9
1 parent cf53ccf
commit 9a8b6e9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -51,9 +51,9 @@ jobs:
       if: ${{ github.event.inputs.test_package == 'true' }}
       run: |
         # 2D test
-        python -c "import numpy; import numpymaxflow; numpymaxflow.maxflow(numpy.zeros(1, 128, 128), numpy.zeros(2, 128, 128), 1.0, 1.0)"
+        python -c "import numpy; import numpymaxflow; numpymaxflow.maxflow(numpy.zeros((1, 128, 128), numpy.float32), numpy.zeros((2, 128, 128), numpy.float32), 1.0, 1.0)"
         # 3D test
-        python -c "import numpy; import numpymaxflow; numpymaxflow.maxflow(numpy.zeros(1, 128, 128, 128), numpy.zeros(2, 128, 128, 128), 1.0, 1.0)"
+        python -c "import numpy; import numpymaxflow; numpymaxflow.maxflow(numpy.zeros((1, 128, 128, 128), numpy.float32), numpy.zeros((2, 128, 128, 128), numpy.float32), 1.0, 1.0)"
       
     - name: Upload Python Dist
       if: ${{ github.event.inputs.save_artifacts == 'true' }}