Importing utils/so3.py takes absolutely forever. Yes, it's a one-time call, but I don't want to suffer it when I move this every time I use this code on a fresh remote VM. So I vectorized the heaviest calculations
I tested this locally, but I don't see a tests/ directory to check this PR automatically.
Also, are you guys open to me adding a few automated tests for this repo? The recent BELKA challenge on kaggle will likely generate a lot of PRs and issues, and having test suite will make approving and checking good changes lot easier.
Importing
utils/so3.py
takes absolutely forever. Yes, it's a one-time call, but I don't want to suffer it when I move this every time I use this code on a fresh remote VM. So I vectorized the heaviest calculationsI tested this locally, but I don't see a
tests/
directory to check this PR automatically.Also, are you guys open to me adding a few automated tests for this repo? The recent BELKA challenge on kaggle will likely generate a lot of PRs and issues, and having test suite will make approving and checking good changes lot easier.