From 6b366d164e7e632408aa184031e9f357ec281871 Mon Sep 17 00:00:00 2001
From: Timothe Jost <timothe.jost@wanadoo.fr>
Date: Thu, 27 Jun 2024 19:05:55 +0200
Subject: [PATCH] get default extra with None in get_run_callbacks
---
src/pypelines/__init__.py | 2 +-
src/pypelines/steps.py | 6 +++++-
2 files changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/pypelines/__init__.py b/src/pypelines/__init__.py
index 2c06cdc..83c651d 100644
--- a/src/pypelines/__init__.py
+++ b/src/pypelines/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.0.65"
+__version__ = "0.0.66"
from . import loggs
from .pipes import *
diff --git a/src/pypelines/steps.py b/src/pypelines/steps.py
index 0fd2bcc..52d9486 100644
--- a/src/pypelines/steps.py
+++ b/src/pypelines/steps.py
@@ -288,7 +288,11 @@ class BaseStep:
return autoload_arguments(loggedmethod(self.generation_mechanism), self)
def get_run_callbacks(self):
- def wrapper(session, extra="", show_plots=True):
+ def wrapper(session, extra=None, show_plots=True):
+
+ if extra is None:
+ extra = self.get_default_extra()
+
logger = logging.getLogger("callback_runner")
for callback_data in self.callbacks:
arguments = {"session": session, "extra": extra, "pipeline": self.pipeline}
--
GitLab