Disable TorchScript compilation when importing operations in the docs

docs/src/conf.py

@@ -5,13 +5,17 @@
 
 import tomli  # Replace by tomllib from std library once docs are build with Python 3.11
 
-import metatrain
-
 
 # When importing metatensor-torch, this will change the definition of the classes
 # to include the documentation
 os.environ["METATENSOR_IMPORT_FOR_SPHINX"] = "1"
 os.environ["RASCALINE_IMPORT_FOR_SPHINX"] = "1"
+# disable TorchScript compilation for metatensor-operations since they will fail with
+# METATENSOR_IMPORT_FOR_SPHINX=1. Remove this when lab-cosmo/metatensor#728 is released.
+os.environ["PYTORCH_JIT"] = "0"
+
+import metatrain  # noqa: E402
+
 
 ROOT = os.path.abspath(os.path.join("..", ".."))