Hook fix

ankushaggarwal · Jan 22, 2023 · 451dfd1 · 451dfd1
1 parent c43c8c8
commit 451dfd1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/gpytorch/means/__init__.py b/gpytorch/means/__init__.py
@@ -10,4 +10,14 @@
 from .multitask_mean import MultitaskMean
 from .zero_mean import ZeroMean
 
-__all__ = ["Mean", "ConstantMean", "ConstantMeanGrad", "ConstantMeanGradGrad", "LinearMean", "LinearMeanGrad", "LinearMeanGradGrad", "MultitaskMean", "ZeroMean"]
+__all__ = [
+    "Mean",
+    "ConstantMean",
+    "ConstantMeanGrad",
+    "ConstantMeanGradGrad",
+    "LinearMean",
+    "LinearMeanGrad",
+    "LinearMeanGradGrad",
+    "MultitaskMean",
+    "ZeroMean",
+]