fix: observations and experiment runner in stevens power law

AutoResearch · Jul 25, 2024 · 68ac55a · 68ac55a
1 parent 891fe7e
commit 68ac55a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/autora/experiment_runner/synthetic/psychophysics/stevens_power_law.py b/src/autora/experiment_runner/synthetic/psychophysics/stevens_power_law.py
@@ -98,7 +98,7 @@ def plotter(
         colors = mcolors.TABLEAU_COLORS
         col_keys = list(colors.keys())
         X = domain()
-        y = ground_truth(X)
+        y = ground_truth(X)[dv1.name]
         plt.plot(X, y, label="Original", c=colors[col_keys[0]])
         if model is not None:
             y = model.predict(X)