AssertionError after warning "Couldn't load cached healpy grid; will fall back to importing healpy"

[kvr2007]

Hello, everyone!

In installed from git repository using pip and then followed the regular tutorial. Particle conversion worked OK, and the training also started. The was a warning:

Couldn't load cached healpy grid; will fall back to importing healpy

when the Epoch -1 initiated, the job crashed with the following error:

  File "/.../python3.9/site-packages/cryodrgnai/so3_grid.py", line 219, in pix2ang_tensor
    assert _GRIDS is not None and n_side in _GRIDS and nest and not lonlat
AssertionError

Somehow I was missing file healpy_grid.json in site-packages and it was not created from scratch.

When I copied healpy_grid.json from github to site-packages/cryodrgnai the job started without any issue.

I think you might need to add healpy_grid.json to [tool.setuptools.package-data] in pyproject.toml so that pip knows what to copy.