Merge branch 'master' into keops_fix

cornellius-gp · Nov 13, 2023 · 3329eec · 3329eec
2 parents f7eaa80 + b491ad6
commit 3329eec
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/gpytorch/lazy/__init__.py b/gpytorch/lazy/__init__.py
@@ -66,11 +66,12 @@ def cat(
 
 
 def __getattr__(name: str) -> Any:
-    warnings.warn(
-        "GPyTorch will be replacing all LazyTensor functionality with the linear operator package. "
-        "Replace all references to gpytorch.lazy.*LazyTensor with linear_operator.operators.*LinearOperator.",
-        DeprecationWarning,
-    )
+    if not name.startswith("_"):
+        warnings.warn(
+            "GPyTorch will be replacing all LazyTensor functionality with the linear operator package. "
+            "Replace all references to gpytorch.lazy.*LazyTensor with linear_operator.operators.*LinearOperator.",
+            DeprecationWarning,
+        )
     if name == "LazyTensor":
         from .lazy_tensor import LazyTensor