fix import

InnerEye/Common/metrics_constants.py

@@ -3,7 +3,6 @@
 # Licensed under the MIT License (MIT). See LICENSE in the repo root for license information.
 # ------------------------------------------------------------------------------------------
 from enum import Enum, unique
-
 # String prefixes when writing training or validation set metrics to a logger
 from typing import Union
 

Tests/ML/test_model_training.py

@@ -12,6 +12,7 @@
 import numpy as np
 import pandas as pd
 import pytest
+from health_ml.utils import BatchTimeCallback
 from torch.utils.data import DataLoader
 
 from InnerEye.Common import fixed_paths