Describe the bug
Plotting should return the figure by default according to the docs, but the default is actually to not return the figure.
Expected behavior
Docs match code.
Additional context
I will change the code to match the docs. Code to change is here