Fix test.

aai-institute · Dec 5, 2023 · 5bc3c8b · 5bc3c8b
1 parent dc2415d
commit 5bc3c8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_sine.py b/tests/test_sine.py
@@ -60,7 +60,7 @@ def test_sine():
             plot_to_tensorboard(writer, "Plot/train", i)
 
         # Test observation
-        obs = dataset._generate_observation(size)
+        obs = dataset.generate_observation(size)
         plt.cla()
         plot_evaluation(model, dataset, obs)
         plot_observation(obs)