add close-figs context on plot directive in ZeroInflatedPoisson class

Author: saurbhc

pymc/distributions/discrete.py

@@ -1469,6 +1469,7 @@ class ZeroInflatedPoisson:
             \end{array} \right.
 
     .. plot::
+        :context: close-figs
 
         import matplotlib.pyplot as plt
         import numpy as np