torch_geometric/nn/pool/select/base.py:65: in <module>
    SelectOutput = torch.jit.script(SelectOutput)
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
/usr/local/lib/python3.11/site-packages/torch/jit/_script.py:1488: in script
    warnings.warn(
E   DeprecationWarning: `torch.jit.script` is deprecated. Please switch to `torch.compile` or `torch.export`.