GFNOrg / torchgfn

GFlowNet library
https://torchgfn.readthedocs.io/en/latest/
Other
209 stars 26 forks source link

`DiscretePolicyEstimator` has no printable representation #134

Closed brunorousseau closed 1 year ago

brunorousseau commented 1 year ago

The class DiscretePolicyEstimator derives from GFNModule. This parent class implements __repr__ by making explicit reference to an environment: Screenshot 2023-08-30 at 3 19 57 PM

The child class inherits this function, but has no env property. Thus, instances of the child class cannot be printed: Screenshot 2023-08-30 at 3 18 54 PM

saleml commented 1 year ago

Thank you for raising the issue. It will be fixed in an imminent release.