code: def clean_context() -> t.Generator[contextlib.ExitStack, None, None]:
      stack = contextlib.ExitStack()
      yield stack
      stack.close()
  
file_path: BentoML/examples/flax/MNIST/tests/conftest.py
imports: from __future__ import annotations
  import contextlib
  import os
  import subprocess
  import sys
  import typing
  from typing import TYPE_CHECKING
  import psutil
  import pytest
  import bentoml
  from bentoml._internal.configuration.containers import BentoMLContainer
  from bentoml.testing.server import host_bento
label_desc: Use image node_id 3 for calling a global function with example usage: clean_context() without return types