Merge pull request openai#57 from marko-bast/fix_legend_draggable_war…

…ning Fix "Legend.draggable() is deprecated" warning
mahdinobar · Jun 28, 2019 · 4625872 · 4625872
2 parents 3a9e27e + e49471a
commit 4625872
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,