Skip to content
Snippets Groups Projects
Commit 9a78349b authored by pswain's avatar pswain
Browse files

feat(pipeline): outputs version of BABY being used

parent bac3de6a
No related branches found
No related tags found
No related merge requests found
...@@ -121,7 +121,8 @@ class BabyRunner(StepABC): ...@@ -121,7 +121,8 @@ class BabyRunner(StepABC):
with_edgemasks=with_edgemasks, with_edgemasks=with_edgemasks,
**kwargs, **kwargs,
) )
return format_segmentation(segmentation, tp) res = format_segmentation(segmentation, tp)
return res
def get_modelset_name_from_params( def get_modelset_name_from_params(
......
...@@ -306,6 +306,10 @@ class Pipeline(ProcessABC): ...@@ -306,6 +306,10 @@ class Pipeline(ProcessABC):
print("\n---\n" + step + "\n---") print("\n---\n" + step + "\n---")
pprint(config[step]) pprint(config[step])
print() print()
try:
print(f"Using Baby {baby.__version}")
except AttributeError:
pass
# extract from configuration # extract from configuration
expt_id = config["general"]["id"] expt_id = config["general"]["id"]
distributed = config["general"]["distributed"] distributed = config["general"]["distributed"]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment