ruff

mehta-lab · Jan 30, 2025 · 91ad800 · 91ad800
1 parent 3f6bc83
commit 91ad800
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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):