code: def pytest_collection_modifyitems(
      session: Session, config: Config, items: list[Item]
  ) -> None:
      try:
          m = bentoml.models.get("mnist_flax")
          print(f"Model exists: {m}")
      except bentoml.exceptions.NotFound:
          subprocess.check_call(
              [
                  sys.executable,
                  f"{os.path.join(PROJECT_DIR, 'train.py')}",
                  "--num-epochs",
                  "2",  # 2 epochs for faster testing
                  "--lr",
                  "0.22",  # speed up training time
                  "--enable-tensorboard",
              ]
          )
  
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 1 for calling a global function with example usage: pytest_collection_modifyitems(session, config, items) without return types