code: def get_total_steps(
      epochs: int, steps_per_epoch: int, train_steps: int
  ):
      """Returns train_steps if provided, otherwise epochs * steps_per_epoch."""
      if train_steps:
          return train_steps
      return epochs * steps_per_epoch
  
imports: import logging
  from collections import defaultdict
  from typing import Dict, List, Tuple, TYPE_CHECKING
  from ludwig.api_annotations import DeveloperAPI
  from ludwig.constants import AUTO, COMBINED, LOSS
  from ludwig.models.base import BaseModel
  from ludwig.modules.metric_modules import get_best_function
  from ludwig.utils.data_utils import save_json
  from ludwig.utils.metric_utils import TrainerMetric
label_desc: Use image node_id 5 for calling a global function with example usage: get_total_steps(epochs, steps_per_epoch, train_steps) and returns: unknown, train_steps