Closed LeonKlein closed 2 years ago
See https://github.com/noegroup/bgflow/blob/ae1dc6a9b3bcd7f0382c98614a572b2ae1b3a540/bgflow/nn/flow/transformer/spline.py#L2 This should be imported in the class. Tests need to be changed accordingly as well.
You are right, this is a bug. It should be an optional dependency though.
I will fix it
See https://github.com/noegroup/bgflow/blob/ae1dc6a9b3bcd7f0382c98614a572b2ae1b3a540/bgflow/nn/flow/transformer/spline.py#L2 This should be imported in the class. Tests need to be changed accordingly as well.