diff --git a/viscy/data/tarrow.py b/viscy/data/tarrow.py
index f554544a..df443581 100644
--- a/viscy/data/tarrow.py
+++ b/viscy/data/tarrow.py
@@ -1,10 +1,11 @@
 from pathlib import Path
+
 import numpy as np
+import torch
 from iohub.ngff import Position, open_ome_zarr
 from lightning.pytorch import LightningDataModule
 from tarrow.data.tarrow_dataset import TarrowDataset
-from torch.utils.data import DataLoader, ConcatDataset
-import torch
+from torch.utils.data import DataLoader
 
 
 class TarrowDataModule(LightningDataModule):
diff --git a/viscy/representation/timearrow.py b/viscy/representation/timearrow.py
index 05c056ae..6333a307 100644
--- a/viscy/representation/timearrow.py
+++ b/viscy/representation/timearrow.py
@@ -1,13 +1,13 @@
 import torch
 import torch.nn as nn
+import torchvision
 from lightning.pytorch import LightningModule
+from lightning.pytorch.callbacks import Callback
 from tarrow.models import TimeArrowNet
 from tarrow.models.losses import DecorrelationLoss
 from torch.optim import Adam
 from torch.optim.lr_scheduler import CyclicLR, ReduceLROnPlateau
-from lightning.pytorch.callbacks import Callback
 from torch.utils.data import DataLoader
-import torchvision
 
 
 class TarrowModule(LightningModule):