From 3e16b20e92056144188879761bf6ae5b12d811a6 Mon Sep 17 00:00:00 2001
From: Théo de la Hogue
Date: Mon, 22 Apr 2024 15:11:35 +0200
Subject: Returning self in PipelineStepEnter decorator wrapper.

---
 src/argaze/DataFeatures.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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
 
-- 
cgit v1.1