re type

GPflow · Oct 9, 2023 · 71f8414 · 71f8414
1 parent 5e0c441
commit 71f8414
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 12 deletions.
diff --git a/gpflow/models/gpr.py b/gpflow/models/gpr.py
@@ -201,7 +201,7 @@ def predict_f(
     def predict_y_faster(
         self,
         Xnew: InputData,
-        posteriors: InputData,
+        posteriors: posteriors.GPRPosterior,
         full_cov: bool = False,
         full_output_cov: bool = False,
     ) -> MeanAndVariance:
@@ -210,8 +210,6 @@ def predict_y_faster(
         which is implenmented based on a given posteior.
         """
 
-        if not isinstance(posteriors, AbstractPosterior):
-            raise ValueError(f"{posteriors} is not a valid gpflow.posteriors")
         f_mean, f_var = posteriors.predict_f(
             Xnew, full_cov=full_cov, full_output_cov=full_output_cov
         )

diff --git a/gpflow/models/sgpr.py b/gpflow/models/sgpr.py
@@ -592,7 +592,7 @@ def predict_f(
     def predict_y_faster(
         self,
         Xnew: InputData,
-        posteriors: InputData,
+        posteriors: posteriors.SGPRPosterior,
         full_cov: bool = False,
         full_output_cov: bool = False,
     ) -> MeanAndVariance:
@@ -602,8 +602,6 @@ def predict_y_faster(
 
         """
 
-        if not isinstance(posteriors, AbstractPosterior):
-            raise ValueError(f"{posteriors} is not a valid gpflow.posteriors")
         f_mean, f_var = posteriors.predict_f(
             Xnew, full_cov=full_cov, full_output_cov=full_output_cov
         )

diff --git a/gpflow/models/svgp.py b/gpflow/models/svgp.py
@@ -266,7 +266,7 @@ def predict_f(
     def predict_y_faster(
         self,
         Xnew: InputData,
-        posteriors: InputData,
+        posteriors: posteriors.BasePosterior,
         full_cov: bool = False,
         full_output_cov: bool = False,
     ) -> MeanAndVariance:
@@ -276,8 +276,6 @@ def predict_y_faster(
 
         """
 
-        if not isinstance(posteriors, AbstractPosterior):
-            raise ValueError(f"{posteriors} is not a valid gpflow.posteriors")
         f_mean, f_var = posteriors.predict_f(
             Xnew, full_cov=full_cov, full_output_cov=full_output_cov
         )

diff --git a/gpflow/models/vgp.py b/gpflow/models/vgp.py
@@ -226,7 +226,7 @@ def predict_f(
     def predict_y_faster(
         self,
         Xnew: InputData,
-        posteriors: InputData,
+        posteriors: posteriors.VGPPosterior,
         full_cov: bool = False,
         full_output_cov: bool = False,
     ) -> MeanAndVariance:
@@ -236,8 +236,6 @@ def predict_y_faster(
 
         """
 
-        if not isinstance(posteriors, AbstractPosterior):
-            raise ValueError(f"{posteriors} is not a valid gpflow.posteriors")
         f_mean, f_var = posteriors.predict_f(
             Xnew, full_cov=full_cov, full_output_cov=full_output_cov
         )