code: def on_training_workload_end(
      self, avg_metrics: Dict[str, Any], batch_metrics: Dict[str, Any]
  ) -> None:
      logging.debug(f"training workload avg_metrics {avg_metrics}")
      logging.debug(f"training workload batch_metrics {batch_metrics}")
      self.training_workloads_ended += 1
  
file_path: determined/harness/tests/experiment/fixtures/pytorch_counter_callback.py
imports: import logging
  from typing import Any, Dict
  from determined import pytorch
label_desc: Use image node_id 9 for calling the Counter obj's underlying member method code with example usage: obj.on_training_workload_end(avg_metrics, batch_metrics) without return types