fix(datasets): Session handling for SnowParkDataset (#1036)

* Fix session handling

Signed-off-by: Merel Theisen <[email protected]>

* Update release notes

Signed-off-by: Merel Theisen <[email protected]>

---------

Signed-off-by: Merel Theisen <[email protected]>

Author: merelcht

kedro-datasets/RELEASE.md

@@ -1,4 +1,4 @@
-# Upcoming Release
+# Upcoming Release 7.0.0
 
 ## Major features and improvements
 
@@ -9,6 +9,7 @@
 
 - Fixed `polars.CSVDataset` `save` method on Windows using `utf-8` as default encoding.
 - Made `table_name` a keyword argument in the `ibis.FileDataset` implementation to be compatible with Ibis 10.0.
+- Fixed how sessions are handled in the `snowflake.SnowflakeTableDataset` implementation.
 
 ## Breaking Changes
 

kedro-datasets/kedro_datasets/snowflake/snowpark_dataset.py

@@ -232,11 +232,7 @@ def load(self) -> DataFrame:
         Returns:
             DataFrame: The loaded data as a Snowpark DataFrame.
         """
-        if self._session is None:
-            raise DatasetError(
-                "No active session. Please initialise a Snowpark session before loading data."
-            )
-        return self._session.table(self._validate_and_get_table_name())
+        return self.session.table(self._validate_and_get_table_name())
 
     def save(self, data: pd.DataFrame | DataFrame) -> None:
         """
@@ -246,12 +242,8 @@ def save(self, data: pd.DataFrame | DataFrame) -> None:
         Args:
             data (pd.DataFrame | DataFrame): The data to save.
         """
-        if self._session is None:
-            raise DatasetError(
-                "No active session. Please initialise a Snowpark session before loading data."
-            )
         if isinstance(data, pd.DataFrame):
-            snowpark_df = self._session.create_dataframe(data)
+            snowpark_df = self.session.create_dataframe(data)
         elif isinstance(data, DataFrame):
             snowpark_df = data
         else:
@@ -270,12 +262,8 @@ def _exists(self) -> bool:
         Returns:
             bool: True if the table exists, False otherwise.
         """
-        if self._session is None:
-            raise DatasetError(
-                "No active session. Please initialise a Snowpark session before loading data."
-            )
         try:
-            self._session.table(
+            self.session.table(
                 f"{self._database}.{self._schema}.{self._table_name}"
             ).show()
             return True