code: def savedmodel_dir(with_version: bool = True) -> str:
      """The directory where we save the SavedModel."""
      model_dir = os.path.join(
          _MODEL_PATH.value,
          _MODEL.value
          + ("" if _MODEL_CLASSIFIER_LAYER.value else "_features"),
      )
      if with_version:
          model_dir = os.path.join(model_dir, str(_MODEL_VERSION.value))
      return model_dir
  
file_path: jax/jax/experimental/jax2tf/examples/saved_model_main.py
imports: import logging
  import os
  from absl import app
  from absl import flags
  from jax.experimental.jax2tf.examples import mnist_lib
  from jax.experimental.jax2tf.examples import saved_model_lib
  import numpy
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 4 for calling a global function with example usage: savedmodel_dir(with_version) and returns: model_dir