noegroup / bgflow

Boltzmann Generators and Normalizing Flows in PyTorch
MIT License
138 stars 36 forks source link

nflows is no longer an optional dependency #49

Closed LeonKlein closed 2 years ago

LeonKlein commented 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.

Olllom commented 2 years ago

You are right, this is a bug. It should be an optional dependency though.

I will fix it