[fix] updating script to work with shape_embed package

uhlmanngroup · Oct 1, 2024 · a3321d8 · a3321d8
1 parent d130dda
commit a3321d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/shape_embed.py b/scripts/shape_embed.py
@@ -33,14 +33,14 @@
 
 torch.multiprocessing.set_sharing_strategy("file_system")
 
-from bioimage_embed import shapes
+from shape_embed import shapes
 import bioimage_embed
 from pytorch_lightning import loggers as pl_loggers
 from torchvision import transforms
 from bioimage_embed.lightning import DataModule
 
 from torchvision import datasets
-from bioimage_embed.shapes.transforms import (
+from shape_embed.shapes.transforms import (
     ImageToCoords,
     CropCentroidPipeline,
     DistogramToCoords,