Merge pull request #61 from SURGroup/bugfix/DimensionReduction

Updated skl_str when using a scikit-learn object
SURGroup · Dec 17, 2020 · dd9f908 · dd9f908
2 parents bfd4505 + 0f62884
commit dd9f908
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/UQpy/DimensionReduction.py b/src/UQpy/DimensionReduction.py
@@ -182,7 +182,7 @@ def __init__(self, distance_method=None, kernel_method=None, interp_object=None,
                 raise TypeError('UQpy: A callable kernel object must be provided.')
 
         # Interpolation.
-        skl_str = "<class 'sklearn.gaussian_process.gpr.GaussianProcessRegressor'>"
+        skl_str = "<class 'sklearn.gaussian_process._gpr.GaussianProcessRegressor'>"
         self.skl = str(type(interp_object))==skl_str
         if interp_object is not None:
             if callable(interp_object) or isinstance(interp_object, Kriging) or self.skl: