From a3321d88dd76b54f7788805308898852e5b46cf0 Mon Sep 17 00:00:00 2001 From: Craig Russell Date: Tue, 1 Oct 2024 10:50:28 +0100 Subject: [PATCH] [fix] updating script to work with shape_embed package --- scripts/shape_embed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/scripts/shape_embed.py b/scripts/shape_embed.py index 73764102..f1064f7c 100644 --- 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,