To solve this issue, you'll need to remove the torch-related dependencies, and move them to the optional dependencies inside pyproject.toml. They're only needed for the RandomLineBO optimizer as far as I remember. You could have poli-baselines[bo] or something like that.