code: def _remove_stage_from_metric_keys(stage, metrics):
      new_metrics = {}
      for metric_name, metric_value in metrics.items():
          new_metrics[
              metric_name.replace(stage + "_", "")
          ] = metric_value
      return new_metrics
  
imports: import os
  import torchnet
  import torch
  from typing import Dict, Any
  import wandb
  from torch.utils.tensorboard import SummaryWriter
  import logging
  from torch_points3d.models import model_interface
label_desc: Use image node_id 10 for calling the BaseTracker obj's underlying member method code with example usage: obj._remove_stage_from_metric_keys(stage, metrics) and returns: new_metrics