search

dtamayo / spock

🖖 A package to determine whether planetary orbital configurations will live long and prosper
GNU General Public License v3.0
64 stars 15 forks source link

Problem with latest version of torch? #21

Closed aobertas closed 1 year ago

aobertas commented 1 year ago

Using python 3.10.2, torch 2.0.0

>>> from spock import FeatureClassifier Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/fs/lustre/scratch/RAID-CITA/obertas/venv-apr-2023-3.10.2/lib/python3.10/site-packages/spock/__init__.py", line 5, in <module> from .deepregressor import DeepRegressor File "/fs/lustre/scratch/RAID-CITA/obertas/venv-apr-2023-3.10.2/lib/python3.10/site-packages/spock/deepregressor.py", line 14, in <module> from .spock_reg_model import load_swag File "/fs/lustre/scratch/RAID-CITA/obertas/venv-apr-2023-3.10.2/lib/python3.10/site-packages/spock/spock_reg_model.py", line 19, in <module> from torch._six import inf ModuleNotFoundError: No module named 'torch._six'

MilesCranmer commented 1 year ago

This should fix it: https://github.com/dtamayo/spock/pull/22. Let me know if it doesn't work,

Thanks! Miles

aobertas commented 1 year ago

Thanks for getting a fix super quickly!