Fixes failing import statements due to kernels restructuring

SURGroup · Mar 7, 2023 · 73b258c · 73b258c
1 parent 0f44a14
commit 73b258c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/unit_tests/sampling/test_adaptive_kriging.py b/tests/unit_tests/sampling/test_adaptive_kriging.py
@@ -1,6 +1,7 @@
 import pytest
 
-from UQpy import GaussianProcessRegression, RBF, LinearRegression
+from UQpy import GaussianProcessRegression, LinearRegression
+from UQpy.utilities.kernels.euclidean_kernels.RBF import RBF
 from UQpy.run_model.model_execution.PythonModel import PythonModel
 from UQpy.utilities.MinimizeOptimizer import MinimizeOptimizer
 

diff --git a/tests/unit_tests/sampling/test_refined_stratified.py b/tests/unit_tests/sampling/test_refined_stratified.py
@@ -1,7 +1,8 @@
 import pytest
 from beartype.roar import BeartypeCallHintPepParamException
 
-from UQpy import GaussianProcessRegression, RBF, LinearRegression
+from UQpy import GaussianProcessRegression, LinearRegression
+from UQpy.utilities.kernels.euclidean_kernels.RBF import RBF
 from UQpy.run_model.model_execution.PythonModel import PythonModel
 from UQpy.utilities.MinimizeOptimizer import MinimizeOptimizer
 from UQpy.sampling.stratified_sampling.refinement.GradientEnhancedRefinement import GradientEnhancedRefinement