be more specific in example

lanl · Nov 7, 2023 · f24e87c · f24e87c
1 parent 5cb429c
commit f24e87c
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)