fix legacy test plot

ourownstory · Nov 10, 2020 · ac60faf · ac60faf
1 parent 6288294
commit ac60faf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_legacy.py b/tests/test_legacy.py
@@ -84,7 +84,7 @@ def test_legacy_ar(self):
             arnet.plot_weights(
                 ar_val=len(ar_params[0]), weights=coeff[0], ar=ar_params[0], save=not self.plot, savedir=results_path
             )
-            arnet.plot_prediction_sample(preds, y, num_obs=100, save=not self.plot, savedir=results_path)
+            arnet.plot_prediction_sample(preds[:100], y[:100], save=not self.plot, savedir=results_path)
             arnet.plot_error_scatter(preds, y, save=not self.plot, savedir=results_path)
 
         if self.save: