Merge pull request #6 from jdtuck/sensitivity

Sensitivity Analysis
lanl · May 8, 2024 · af2caad · af2caad
2 parents e9c6804 + f24e87c
commit af2caad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_bassPCA_fit.py b/tests/test_bassPCA_fit.py
@@ -5,7 +5,7 @@
 def test_bassPCA_fit():
     # Friedman function with functional response
     def f2(x):
-        out = 10. * np.sin(np.pi * tt * x[1]) + 20. * (x[2] - .5) ** 2 + 10 * x[3] + 5. * x[4]
+        out = 10. * np.sin(np.pi * np.linspace(0, 1, 50) * x[1]) + 20. * (x[2] - .5) ** 2 + 10 * x[3] + 5. * x[4]
         return out
 
     np.random.seed(0)