search

GFNOrg / torchgfn

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

checks whether user-defined env.step method returns the expected type #165

Closed josephdviviano closed 6 months ago

josephdviviano commented 6 months ago

This implements a check to ensure that any user-defined env.step() call returns a torch.Tensor instance, not a States instance.