From 9a78349b7e6b1ef6948dccc74d363481f1278cf8 Mon Sep 17 00:00:00 2001
From: Peter Swain <peter.swain@ed.ac.uk>
Date: Wed, 6 Dec 2023 09:57:58 +0000
Subject: [PATCH] feat(pipeline): outputs version of BABY being used
---
src/aliby/baby_sitter.py | 3 ++-
src/aliby/pipeline.py | 4 ++++
2 files changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/aliby/baby_sitter.py b/src/aliby/baby_sitter.py
index 2b56c6f..a7d58df 100644
--- a/src/aliby/baby_sitter.py
+++ b/src/aliby/baby_sitter.py
@@ -121,7 +121,8 @@ class BabyRunner(StepABC):
with_edgemasks=with_edgemasks,
**kwargs,
)
- return format_segmentation(segmentation, tp)
+ res = format_segmentation(segmentation, tp)
+ return res
def get_modelset_name_from_params(
diff --git a/src/aliby/pipeline.py b/src/aliby/pipeline.py
index 6eaeaa1..fa52f6b 100644
--- a/src/aliby/pipeline.py
+++ b/src/aliby/pipeline.py
@@ -306,6 +306,10 @@ class Pipeline(ProcessABC):
print("\n---\n" + step + "\n---")
pprint(config[step])
print()
+ try:
+ print(f"Using Baby {baby.__version}")
+ except AttributeError:
+ pass
# extract from configuration
expt_id = config["general"]["id"]
distributed = config["general"]["distributed"]
--
GitLab