aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/argaze/DataFeatures.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/argaze/DataFeatures.py b/src/argaze/DataFeatures.py
index 1bc43a8..6f08f30 100644
--- a/src/argaze/DataFeatures.py
+++ b/src/argaze/DataFeatures.py
@@ -680,7 +680,9 @@ def PipelineStepEnter(method):
PipelineStepObject.__enter__(self)
- return method(self)
+ method(self)
+
+ return self
return wrapper