Fix imports

computational-cell-analytics · Dec 16, 2023 · 22d830b · 22d830b
1 parent 0db52d4
commit 22d830b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/micro_sam/training/medsam_trainer.py b/micro_sam/training/medsam_trainer.py
@@ -1,7 +1,9 @@
+import os
 import time
 
 import torch
 from torchvision.utils import make_grid
+
 import torch_em
 from torch_em.trainer.logger_base import TorchEmLogger