Move figures from .data to .utils

map2map/train.py

@@ -14,10 +14,9 @@ from torch.utils.data import DataLoader
 from torch.utils.tensorboard import SummaryWriter
 
 from .data import FieldDataset, DistFieldSampler
-from .data.figures import plt_slices, plt_power
 from . import models
 from .models import narrow_cast, resample, lag2eul
-from .utils import import_attr, load_model_state_dict
+from .utils import import_attr, load_model_state_dict, plt_slices, plt_power
 
 
 ckpt_link = 'checkpoint.pt'

map2map/utils/__init__.py

@@ -1,2 +1,5 @@
 from .imp import import_attr
+
 from .state import load_model_state_dict
+
+from .figures import plt_slices, plt_power