It's probably not a bad idea to *optionally* make it a context manager, but I don't think it should be required.  The imposes a level of structure that's beyond the quick-and-dirty plotting scripts that most people write.