build: fix packages versions and scikit-version

Packages requiere newer Python versions than 3.8.
But some packages are old and use deprecated features (scikit-video),
so I added a quick fix. Ideally I fix it at some point.
Some versions on requirements are also more clearly specified, as there
are other deprecations.

.github/workflows/integration_test.yml

@@ -14,10 +14,10 @@ jobs:
     steps:
       - uses: actions/checkout@v2
 
-      - name: Set up Python 3.8
+      - name: Set up Python 3.10
         uses: actions/setup-python@v2
         with:
-          python-version: '3.8'
+          python-version: '3.10'
           architecture: 'x64'
 
       - name: apt-get

examples/detect.py

@@ -2,6 +2,12 @@
 import deeptangle as dt
 import matplotlib.pyplot as plt
 from skimage.exposure import equalize_adapthist
+import numpy
+
+# scikit-video uses deprecated numpy.float, numpy.int
+# hacky fix: https://github.com/scikit-video/scikit-video/issues/154
+numpy.float = numpy.float64
+numpy.int = numpy.int_
 import skvideo.io
 
 

examples/track.py

@@ -9,6 +9,12 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from skimage.exposure import equalize_adapthist
+
+# scikit-video uses deprecated numpy.float, numpy.int
+# hacky fix: https://github.com/scikit-video/scikit-video/issues/154
+import numpy
+numpy.float = numpy.float64
+numpy.int = numpy.int_
 import skvideo.io
 
 import deeptangle as dt

requirements.txt

@@ -3,11 +3,12 @@ scikit-image
 scikit-video
 optax
 chex
-jax
+jax>=0.4.16
+jaxlib>=0.4.20
 dm-pix
 scikit-learn
-numpy==1.21.6
-numba==0.55
+numpy>=1.21.6
+numba>=0.55
 matplotlib
-https://github.com/alonfnt/dm-haiku/archive/refs/heads/avg_pool_perf.zip
+dm-haiku
 trackpy