Merge pull request #57 from marko-bast/fix_legend_draggable_warning

Fix "Legend.draggable() is deprecated" warning
openai · Jun 28, 2019 · 5267f1a · 5267f1a
2 parents da53ce3 + ecf9bc0
commit 5267f1a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spinup/utils/plot.py b/spinup/utils/plot.py
@@ -39,7 +39,7 @@ def plot_data(data, xaxis='Epoch', value="AverageEpRet", condition="Condition1",
 
     Changes the colorscheme and the default legend style, though.
     """
-    plt.legend(loc='best').draggable()
+    plt.legend(loc='best').set_draggable(True)
 
     """
     For the version of the legend used in the Spinning Up benchmarking page,