State that array_context_for_pickling is a context manager