torchgfn
torchgfn copied to clipboard
improve box_utils.py file
Copied from https://github.com/saleml/torchgfn/pull/66
- Simplify the code. I don't think we need the BoxPFEstimator and BoxPFNeuralNet to be two distinct classes. It would be easier if all the logical existed in the BoxPFEstimator. But we can debate this.
- Ensure the Tabular case works as intended.
- Put the tests in a proper place. But where?