Fix numba changes and update python and pytorch version
Update to python 3.8 and pytorch 1.5 (#36) * Update to python 1.5 * Update setup.py * Bump version: 0.2.0 → 0.2.1 Co-authored-by: havakv <haavard.kvamme@gmail..com>
Update to python 3.8 and pytorch 1.5 (#36) * Update to python 1.5 * Update setup.py * Bump version: 0.2.0 → 0.2.1 Co-authored-by: havakv <haavard.kvamme@gmail..com>