Coverage for geometric_kernels/torch.py: 100%

1""" 

2Loads PyTorch backend in lab, spherical_harmonics and geometric_kernels. 

3 

4..note:: 

5 A tutorial on the JAX backend is available in the 

6 :doc:`backends/PyTorch_Graph.ipynb </examples/backends/PyTorch_Graph>` 

7 notebook. 

8""" 

9 

10import logging 

11 

12import lab.torch # noqa 

13import spherical_harmonics.torch # noqa 

14 

15import geometric_kernels.lab_extras.torch # noqa 

16 

17logging.getLogger(__name__).info("Torch backend enabled.")