From 9d61fed068fc18c6657f3dfa9d58fcb5e6b42a1f Mon Sep 17 00:00:00 2001
From: Rahul Dubey <100185371+RahulDubey391@users.noreply.github.com>
Date: Sun, 17 Dec 2023 16:46:37 +0530
Subject: [PATCH] Adding Method inverse_transform() signature

---
 snowflake/ml/modeling/framework/base.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/snowflake/ml/modeling/framework/base.py b/snowflake/ml/modeling/framework/base.py
index 4c03ef99..5f14ce0d 100644
--- a/snowflake/ml/modeling/framework/base.py
+++ b/snowflake/ml/modeling/framework/base.py
@@ -511,6 +511,14 @@ def transform(self, dataset: snowpark.DataFrame) -> snowpark.DataFrame:
     def transform(self, dataset: pd.DataFrame) -> pd.DataFrame:
         ...
 
+    @overload
+    def inverse_transform(self, dataset: snowpark.DataFrame) -> snowpark.DataFrame:
+        ...
+        
+    @overload
+    def inverse_transform(self, dataset: pd.DataFrame) -> pd.DataFrame:
+        ...
+
     @abstractmethod
     def transform(self, dataset: Union[snowpark.DataFrame, pd.DataFrame]) -> Union[snowpark.DataFrame, pd.DataFrame]:
         raise NotImplementedError()