Update __init__.py (#2423)

cornellius-gp · Oct 26, 2023 · b491ad6 · b491ad6
1 parent 43383c2
commit b491ad6
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