From f45bbe196bab64406196042aa99527223e298135 Mon Sep 17 00:00:00 2001
From: DrozdovMK <drozdovmk@ioes.ru>
Date: Wed, 2 Jul 2025 15:02:06 +0300
Subject: [PATCH] Add set_output method in FeatureAugmenter for better
 compability with sklearn pipelines

---
 tsfresh/transformers/feature_augmenter.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/tsfresh/transformers/feature_augmenter.py b/tsfresh/transformers/feature_augmenter.py
index 421177d1..928d5c44 100644
--- a/tsfresh/transformers/feature_augmenter.py
+++ b/tsfresh/transformers/feature_augmenter.py
@@ -232,3 +232,12 @@ def transform(self, X):
         )
 
         return X
+    
+    def set_output(self, *, transform = None):
+        """
+        Set the output container format of the transform method.
+        Since it is assumed that the transformer output goes to the pd.DataFrame,
+        nothing is done (the method was created for compatibility with sklearn)
+        For example when calling sklearn.pipeline.FeatureUnion.set_output(transform="pandas")
+        """
+        return self