From 33e9c9f29260f9cf05d7c6a5f9c7201287eddfda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <amuoz@ed.ac.uk>
Date: Wed, 29 Jun 2022 18:20:02 +0100
Subject: [PATCH] fix(pipeline): Integrate get_image_class
---
aliby/pipeline.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/aliby/pipeline.py b/aliby/pipeline.py
index 0e175f4a..7f02704b 100644
--- a/aliby/pipeline.py
+++ b/aliby/pipeline.py
@@ -38,7 +38,7 @@ from tqdm import tqdm
from aliby.baby_client import BabyParameters, BabyRunner
from aliby.haystack import initialise_tf
from aliby.io.dataset import Dataset, DatasetLocal
-from aliby.io.image import Image, ImageLocal
+from aliby.io.image import get_image_class
from aliby.tile.tiler import Tiler, TilerParameters
from extraction.core.extractor import Extractor, ExtractorParameters
from extraction.core.functions.defaults import exparams_from_meta
@@ -361,7 +361,7 @@ class Pipeline(ProcessABC):
# Set up
directory = general_config["directory"]
- image_wrapper = ImageLocal if isinstance(image_id, str) else Image
+ image_wrapper = get_image_class(image_id)
with image_wrapper(
image_id, **self.general.get("server_info", {})
--
GitLab