code: def test_history_epoch_column(self, history, ref):
      total_losses = history[:, "total_loss"]
      total_losses_ref = [n["total_loss"] for n in ref]
      assert total_losses == total_losses_ref
  
imports: import pickle
  import time
  import multiprocessing
  from functools import partial
  import numpy
  import pytest
  import torch
  from torch.distributed import TCPStore
  from sklearn.datasets import make_classification
label_desc: Use image node_id 6 for calling the TestHistory obj's underlying member method code with example usage: obj.test_history_epoch_column(history, ref) without return types