UserWarning from torch.utils.checkpoint

[DhanshreeA]

I am running tabpfn v0.0.9 within a Python 3.10 environment, and torch version 2.2.1. There's the following user warning:

/usr/local/lib/python3.10/dist-packages/torch/utils/checkpoint.py:460: UserWarning: torch.utils.checkpoint: please pass in use_reentrant=True or use_reentrant=False explicitly. The default value of use_reentrant will be updated to be False in the future. To maintain current behavior, pass use_reentrant=True. It is recommended that you use use_reentrant=False. Refer to docs for more details on the differences between the two variants.
  warnings.warn(

This is coming from transformer_prediction_interface.py which makes use of the checkpoints function.