From d9f610eab87f7ca58a2832e9bc305ee2271cc07b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <amuoz@ed.ac.uk>
Date: Mon, 20 Dec 2021 12:34:02 +0000
Subject: [PATCH] allow to pass ints as strains
---
aliby/pipeline.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/aliby/pipeline.py b/aliby/pipeline.py
index 73cd003e..b1d75dae 100644
--- a/aliby/pipeline.py
+++ b/aliby/pipeline.py
@@ -145,7 +145,10 @@ class Pipeline(ProcessABC):
config["general"]["directory"] = directory
# Filter TODO integrate filter onto class and add regex
- image_ids = {k: v for k, v in image_ids.items() if k.startswith(strain_filter)}
+ if isinstance(strain_filter, str):
+ image_ids = {k: v for k, v in image_ids.items() if k.startswith(strain_filter)}
+ elif isinstance(strain_filter, int):
+ image_ids = {k:v for i,( k,v ) in enumerate( image_ids.items() ) if i==0}
if distributed != 0: # Gives the number of simultaneous processes
with Pool(distributed) as p:
--
GitLab