[fix] updating script to work with shape_embed package

uhlmanngroup · Oct 1, 2024 · 5669be7 · 5669be7
1 parent 0055bcd
commit 5669be7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/scripts/_shape_embed.py b/scripts/_shape_embed.py
@@ -23,7 +23,7 @@
 
 torch.multiprocessing.set_sharing_strategy("file_system")
 
-from bioimage_embed import shapes
+from shape_embed import shapes
 import bioimage_embed
 
 # Note - you must have torchvision installed for this example

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,