code: def get_model_dirs(gcs_bucket, job_name):
      gcs_base_path = f"gs://{gcs_bucket}/{job_name}"
      return {
          "tensorboard_logs": os.path.join(gcs_base_path, "logs"),
          "model_checkpoint": os.path.join(
              gcs_base_path, "checkpoints"
          ),
          "saved_model": os.path.join(gcs_base_path, "saved_model"),
      }
  
imports: import os
  import pickle
  from typing import Any, Dict, Optional
  import uuid
  from ..None import machine_config
  from ..None import run
  import tensorflow
  import tensorflow_datasets
  from official.legacy.image_classification.efficientnet import efficientnet_model
  from official.legacy.image_classification.resnet import resnet_model
label_desc: Use image node_id 2 for calling a global function with example usage: get_model_dirs(gcs_bucket, job_name) and returns: dict