Rename torch.Tensor to Tensor in docs

cornellius-gp · Jul 2, 2024 · 5d1497d · 5d1497d
1 parent 9ff4640
commit 5d1497d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -253,6 +253,8 @@ def _convert_internal_and_external_class_to_strings(annotation):
     module = annotation.__module__ + "."
     if module.split(".")[0] == "gpytorch":
         module = "~" + module
+    elif module == "torch.":
+        module = "~torch."
     elif module == "linear_operator.operators._linear_operator.":
         module = "~linear_operator."
     elif module == "builtins.":