snowpark-checkpoints-validators 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

snowflake/snowpark_checkpoints/__version__.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.2.1"
+__version__ = "0.3.0"

snowflake/snowpark_checkpoints/io_utils/__init__.py

@@ -0,0 +1,26 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__all__ = ["EnvStrategy", "IOFileManager", "IODefaultStrategy"]
+
+from snowflake.snowpark_checkpoints.io_utils.io_env_strategy import (
+    EnvStrategy,
+)
+from snowflake.snowpark_checkpoints.io_utils.io_default_strategy import (
+    IODefaultStrategy,
+)
+from snowflake.snowpark_checkpoints.io_utils.io_file_manager import (
+    IOFileManager,
+)

snowflake/snowpark_checkpoints/io_utils/io_default_strategy.py

@@ -0,0 +1,57 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import glob
+import os
+
+from pathlib import Path
+from typing import Optional
+
+from snowflake.snowpark_checkpoints.io_utils import EnvStrategy
+
+
+class IODefaultStrategy(EnvStrategy):
+    def mkdir(self, path: str, exist_ok: bool = False) -> None:
+        os.makedirs(path, exist_ok=exist_ok)
+
+    def folder_exists(self, path: str) -> bool:
+        return os.path.isdir(path)
+
+    def file_exists(self, path: str) -> bool:
+        return os.path.isfile(path)
+
+    def write(self, file_path: str, file_content: str, overwrite: bool = True) -> None:
+        mode = "w" if overwrite else "x"
+        with open(file_path, mode) as file:
+            file.write(file_content)
+
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        with open(file_path, mode=mode, encoding=encoding) as file:
+            return file.read()
+
+    def read_bytes(self, file_path: str) -> bytes:
+        with open(file_path, mode="rb") as f:
+            return f.read()
+
+    def ls(self, path: str, recursive: bool = False) -> list[str]:
+        return glob.glob(path, recursive=recursive)
+
+    def getcwd(self) -> str:
+        return os.getcwd()
+
+    def telemetry_path_files(self, path: str) -> Path:
+        return Path(path)

snowflake/snowpark_checkpoints/io_utils/io_env_strategy.py

@@ -0,0 +1,133 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from abc import ABC, abstractmethod
+from pathlib import Path
+from typing import Optional
+
+
+class EnvStrategy(ABC):
+
+    """An abstract base class that defines methods for file and directory operations.
+
+    Subclasses should implement these methods to provide environment-specific behavior.
+    """
+
+    @abstractmethod
+    def mkdir(self, path: str, exist_ok: bool = False) -> None:
+        """Create a directory.
+
+        Args:
+            path: The name of the directory to create.
+            exist_ok: If False, an error is raised if the directory already exists.
+
+        """
+
+    @abstractmethod
+    def folder_exists(self, path: str) -> bool:
+        """Check if a folder exists.
+
+        Args:
+            path: The path to the folder.
+
+        Returns:
+            bool: True if the folder exists, False otherwise.
+
+        """
+
+    @abstractmethod
+    def file_exists(self, path: str) -> bool:
+        """Check if a file exists.
+
+        Args:
+            path: The path to the file.
+
+        Returns:
+            bool: True if the file exists, False otherwise.
+
+        """
+
+    @abstractmethod
+    def write(self, file_path: str, file_content: str, overwrite: bool = True) -> None:
+        """Write content to a file.
+
+        Args:
+            file_path: The name of the file to write to.
+            file_content: The content to write to the file.
+            overwrite: If True, overwrite the file if it exists.
+
+        """
+
+    @abstractmethod
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        """Read content from a file.
+
+        Args:
+            file_path: The path to the file to read from.
+            mode: The mode in which to open the file.
+            encoding: The encoding to use for reading the file.
+
+        Returns:
+            str: The content of the file.
+
+        """
+
+    @abstractmethod
+    def read_bytes(self, file_path: str) -> bytes:
+        """Read binary content from a file.
+
+        Args:
+            file_path: The path to the file to read from.
+
+        Returns:
+            bytes: The binary content of the file.
+
+        """
+
+    @abstractmethod
+    def ls(self, path: str, recursive: bool = False) -> list[str]:
+        """List the contents of a directory.
+
+        Args:
+            path: The path to the directory.
+            recursive: If True, list the contents recursively.
+
+        Returns:
+            list[str]: A list of the contents of the directory.
+
+        """
+
+    @abstractmethod
+    def getcwd(self) -> str:
+        """Get the current working directory.
+
+        Returns:
+            str: The current working directory.
+
+        """
+
+    @abstractmethod
+    def telemetry_path_files(self, path: str) -> Path:
+        """Get the path to the telemetry files.
+
+        Args:
+            path: The path to the telemetry directory.
+
+        Returns:
+            Path: The path object representing the telemetry files.
+
+        """

snowflake/snowpark_checkpoints/io_utils/io_file_manager.py

@@ -0,0 +1,76 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+# http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from pathlib import Path
+from typing import Optional
+
+from snowflake.snowpark_checkpoints.io_utils import (
+    EnvStrategy,
+    IODefaultStrategy,
+)
+from snowflake.snowpark_checkpoints.singleton import Singleton
+
+
+class IOFileManager(metaclass=Singleton):
+    def __init__(self, strategy: Optional[EnvStrategy] = None):
+        self.strategy = strategy or IODefaultStrategy()
+
+    def mkdir(self, path: str, exist_ok: bool = False) -> None:
+        return self.strategy.mkdir(path, exist_ok)
+
+    def folder_exists(self, path: str) -> bool:
+        return self.strategy.folder_exists(path)
+
+    def file_exists(self, path: str) -> bool:
+        return self.strategy.file_exists(path)
+
+    def write(self, file_path: str, file_content: str, overwrite: bool = True) -> None:
+        return self.strategy.write(file_path, file_content, overwrite)
+
+    def read(
+        self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+    ) -> str:
+        return self.strategy.read(file_path, mode, encoding)
+
+    def read_bytes(self, file_path: str) -> bytes:
+        return self.strategy.read_bytes(file_path)
+
+    def ls(self, path: str, recursive: bool = False) -> list[str]:
+        return self.strategy.ls(path, recursive)
+
+    def getcwd(self) -> str:
+        return self.strategy.getcwd()
+
+    def telemetry_path_files(self, path: str) -> Path:
+        return self.strategy.telemetry_path_files(path)
+
+    def set_strategy(self, strategy: EnvStrategy):
+        """Set the strategy for file and directory operations.
+
+        Args:
+            strategy (EnvStrategy): The strategy to use for file and directory operations.
+
+        """
+        self.strategy = strategy
+
+
+def get_io_file_manager():
+    """Get the singleton instance of IOFileManager.
+
+    Returns:
+        IOFileManager: The singleton instance of IOFileManager.
+
+    """
+    return IOFileManager()

snowflake/snowpark_checkpoints/utils/extra_config.py

@@ -18,6 +18,7 @@ import os
 
 from typing import Optional
 
+from snowflake.snowpark_checkpoints.io_utils.io_file_manager import get_io_file_manager
 from snowflake.snowpark_checkpoints.utils.constants import (
     SNOWFLAKE_CHECKPOINT_CONTRACT_FILE_PATH_ENV_VAR,
 )
@@ -28,7 +29,48 @@ LOGGER = logging.getLogger(__name__)
 
 # noinspection DuplicatedCode
 def _get_checkpoint_contract_file_path() -> str:
-    return os.environ.get(SNOWFLAKE_CHECKPOINT_CONTRACT_FILE_PATH_ENV_VAR, os.getcwd())
+    return os.environ.get(
+        SNOWFLAKE_CHECKPOINT_CONTRACT_FILE_PATH_ENV_VAR, get_io_file_manager().getcwd()
+    )
+
+
+def _set_conf_io_strategy() -> None:
+    try:
+        from snowflake.snowpark_checkpoints.io_utils.io_default_strategy import (
+            IODefaultStrategy,
+        )
+        from snowflake.snowpark_checkpoints_configuration.io_utils.io_file_manager import (
+            EnvStrategy as ConfEnvStrategy,
+        )
+        from snowflake.snowpark_checkpoints_configuration.io_utils.io_file_manager import (
+            get_io_file_manager as get_conf_io_file_manager,
+        )
+
+        is_default_strategy = isinstance(
+            get_io_file_manager().strategy, IODefaultStrategy
+        )
+
+        if is_default_strategy:
+            return
+
+        class CustomConfEnvStrategy(ConfEnvStrategy):
+            def file_exists(self, path: str) -> bool:
+                return get_io_file_manager().file_exists(path)
+
+            def read(
+                self, file_path: str, mode: str = "r", encoding: Optional[str] = None
+            ) -> Optional[str]:
+                return get_io_file_manager().read(file_path, mode, encoding)
+
+            def getcwd(self) -> str:
+                return get_io_file_manager().getcwd()
+
+        get_conf_io_file_manager().set_strategy(CustomConfEnvStrategy())
+
+    except ImportError:
+        LOGGER.debug(
+            "snowpark-checkpoints-configuration is not installed. Cannot get a checkpoint metadata instance."
+        )
 
 
 # noinspection DuplicatedCode
@@ -39,6 +81,7 @@ def _get_metadata():
         )
 
         path = _get_checkpoint_contract_file_path()
+        _set_conf_io_strategy()
         LOGGER.debug("Loading checkpoint metadata from '%s'", path)
         metadata = CheckpointMetadata(path)
         return True, metadata

snowflake/snowpark_checkpoints/utils/telemetry.py

@@ -30,16 +30,34 @@ from sys import platform
 from typing import Any, Callable, Optional, TypeVar
 from uuid import getnode
 
-from snowflake.connector import (
-    SNOWFLAKE_CONNECTOR_VERSION,
-    time_util,
-)
-from snowflake.connector.constants import DIRS as SNOWFLAKE_DIRS
-from snowflake.connector.network import SnowflakeRestful
+from snowflake.connector.description import PLATFORM as CONNECTOR_PLATFORM
 from snowflake.connector.telemetry import TelemetryClient
 from snowflake.snowpark import VERSION as SNOWPARK_VERSION
 from snowflake.snowpark import dataframe as snowpark_dataframe
 from snowflake.snowpark.session import Session
+from snowflake.snowpark_checkpoints.io_utils.io_file_manager import (
+    get_io_file_manager,
+)
+
+
+try:
+    """
+    The following imports are used to log telemetry events in the Snowflake Connector.
+    """
+    from snowflake.connector import (
+        SNOWFLAKE_CONNECTOR_VERSION,
+        time_util,
+    )
+    from snowflake.connector.constants import DIRS as SNOWFLAKE_DIRS
+    from snowflake.connector.network import SnowflakeRestful
+except Exception:
+    """
+    Set default import values for the Snowflake Connector when using snowpark-checkpoints in stored procedures.
+    """
+    SNOWFLAKE_CONNECTOR_VERSION = ""
+    time_util = None
+    SNOWFLAKE_DIRS = ""
+    SnowflakeRestful = None
 
 
 try:
@@ -92,7 +110,7 @@ class TelemetryManager(TelemetryClient):
             path: path to write telemetry.
 
         """
-        os.makedirs(path, exist_ok=True)
+        get_io_file_manager().mkdir(str(path), exist_ok=True)
         self.sc_folder_path = path
 
     def sc_log_error(
@@ -200,7 +218,7 @@ class TelemetryManager(TelemetryClient):
 
         """
         try:
-            os.makedirs(self.sc_folder_path, exist_ok=True)
+            get_io_file_manager().mkdir(str(self.sc_folder_path), exist_ok=True)
             for event in batch:
                 message = event.get("message")
                 if message is not None:
@@ -210,8 +228,7 @@ class TelemetryManager(TelemetryClient):
                         f'_telemetry_{message.get("type")}.json'
                     )
                     json_content = self._sc_validate_folder_space(event)
-                    with open(file_path, "w") as json_file:
-                        json_file.write(json_content)
+                    get_io_file_manager().write(str(file_path), json_content)
         except Exception:
             pass
 
@@ -238,10 +255,10 @@ class TelemetryManager(TelemetryClient):
         if not self.sc_is_enabled or self.sc_is_testing or not self._rest:
             return
         batch = []
-        for file in self.sc_folder_path.glob("*.json"):
-            with open(file) as json_file:
-                data_dict = json.load(json_file)
-                batch.append(data_dict)
+        for file in get_io_file_manager().ls(f"{self.sc_folder_path}/*.json"):
+            json_content = get_io_file_manager().read(file)
+            data_dict = json.loads(json_content)
+            batch.append(data_dict)
         if batch == []:
             return
         body = {"logs": batch}
@@ -253,14 +270,17 @@ class TelemetryManager(TelemetryClient):
             timeout=5,
         )
         if ret.get("success"):
-            for file in self.sc_folder_path.glob("*.json"):
+            for file_path in get_io_file_manager().ls(f"{self.sc_folder_path}/*.json"):
+                file = get_io_file_manager().telemetry_path_files(file_path)
                 file.unlink()
 
     def _sc_is_telemetry_testing(self) -> bool:
         is_testing = os.getenv("SNOWPARK_CHECKPOINTS_TELEMETRY_TESTING") == "true"
         if is_testing:
             local_telemetry_path = (
-                Path(os.getcwd()) / "snowpark-checkpoints-output" / "telemetry"
+                Path(get_io_file_manager().getcwd())
+                / "snowpark-checkpoints-output"
+                / "telemetry"
             )
             self.set_sc_output_path(local_telemetry_path)
             self.sc_is_enabled = True
@@ -359,7 +379,7 @@ def _get_metadata() -> dict:
     }
 
 
-def _get_version() -> str:
+def _get_version() -> Optional[str]:
     """Get the version of the package.
 
     Returns:
@@ -370,11 +390,10 @@ def _get_version() -> str:
         directory_levels_up = 1
         project_root = Path(__file__).resolve().parents[directory_levels_up]
         version_file_path = project_root / VERSION_FILE_NAME
-        with open(version_file_path) as file:
-            content = file.read()
-            version_match = re.search(VERSION_VARIABLE_PATTERN, content, re.MULTILINE)
-            if version_match:
-                return version_match.group(1)
+        content = get_io_file_manager().read(str(version_file_path))
+        version_match = re.search(VERSION_VARIABLE_PATTERN, content, re.MULTILINE)
+        if version_match:
+            return version_match.group(1)
         return None
     except Exception:
         return None
@@ -390,7 +409,10 @@ def _get_folder_size(folder_path: Path) -> int:
         int: The size of the folder in bytes.
 
     """
-    return sum(f.stat().st_size for f in folder_path.glob("*.json") if f.is_file())
+    sum_size = 0
+    for f in get_io_file_manager().ls(f"{folder_path}/*.json"):
+        sum_size += get_io_file_manager().telemetry_path_files(f).stat().st_size
+    return sum_size
 
 
 def _free_up_space(folder_path: Path, max_size: int) -> None:
@@ -401,9 +423,13 @@ def _free_up_space(folder_path: Path, max_size: int) -> None:
         max_size (int): The maximum allowed size of the folder in bytes.
 
     """
-    files = sorted(folder_path.glob("*.json"), key=lambda f: f.stat().st_mtime)
+    files = sorted(
+        get_io_file_manager().ls(f"{folder_path}/*.json"),
+        key=lambda f: f.stat().st_mtime,
+    )
     current_size = _get_folder_size(folder_path)
-    for file in files:
+    for file_path in files:
+        file = get_io_file_manager().telemetry_path_files(file_path)
         if current_size <= max_size:
             break
         current_size -= file.stat().st_size
@@ -482,12 +508,22 @@ def get_load_json(json_schema: str) -> dict:
 
     """
     try:
-        with open(json_schema, encoding="utf-8") as file:
-            return json.load(file)
+        file_content = get_io_file_manager().read(json_schema, encoding="utf-8")
+        return json.loads(file_content)
     except (OSError, json.JSONDecodeError) as e:
         raise ValueError(f"Error reading JSON schema file: {e}") from None
 
 
+def _is_in_stored_procedure() -> bool:
+    """Check if the code is running in a stored procedure.
+
+    Returns:
+        bool: True if the code is running in a stored procedure, False otherwise.
+
+    """
+    return CONNECTOR_PLATFORM == "XP"
+
+
 def extract_parameters(
     func: Callable, args: tuple, kwargs: dict, params_list: Optional[list[str]]
 ) -> dict:
@@ -835,7 +871,10 @@ def report_telemetry(
             except Exception as err:
                 func_exception = err
 
-            if os.getenv("SNOWPARK_CHECKPOINTS_TELEMETRY_ENABLED") == "false":
+            if (
+                os.getenv("SNOWPARK_CHECKPOINTS_TELEMETRY_ENABLED") == "false"
+                or _is_in_stored_procedure()
+            ):
                 return result
             telemetry_event = None
             data = None

snowflake/snowpark_checkpoints/utils/utils_checks.py

@@ -28,6 +28,7 @@ from pandera import DataFrameSchema
 
 from snowflake.snowpark import DataFrame as SnowparkDataFrame
 from snowflake.snowpark_checkpoints.errors import SchemaValidationError
+from snowflake.snowpark_checkpoints.io_utils.io_file_manager import get_io_file_manager
 from snowflake.snowpark_checkpoints.job_context import SnowparkJobContext
 from snowflake.snowpark_checkpoints.snowpark_sampler import (
     SamplingAdapter,
@@ -154,13 +155,15 @@ def _generate_schema(
     LOGGER.info(
         "Generating Pandera DataFrameSchema for checkpoint: '%s'", checkpoint_name
     )
-    current_directory_path = output_path if output_path else os.getcwd()
+    current_directory_path = (
+        output_path if output_path else get_io_file_manager().getcwd()
+    )
 
     output_directory_path = os.path.join(
         current_directory_path, SNOWPARK_CHECKPOINTS_OUTPUT_DIRECTORY_NAME
     )
 
-    if not os.path.exists(output_directory_path):
+    if not get_io_file_manager().folder_exists(output_directory_path):
         raise ValueError(
             """Output directory snowpark-checkpoints-output does not exist.
 Please run the Snowpark checkpoint collector first."""
@@ -171,14 +174,14 @@ Please run the Snowpark checkpoint collector first."""
         CHECKPOINT_JSON_OUTPUT_FILE_FORMAT_NAME.format(checkpoint_name),
     )
 
-    if not os.path.exists(checkpoint_schema_file_path):
+    if not get_io_file_manager().file_exists(checkpoint_schema_file_path):
         raise ValueError(
             f"Checkpoint {checkpoint_name} JSON file not found. Please run the Snowpark checkpoint collector first."
         )
 
     LOGGER.info("Reading schema from file: '%s'", checkpoint_schema_file_path)
-    with open(checkpoint_schema_file_path) as schema_file:
-        checkpoint_schema_config = json.load(schema_file)
+    schema_file = get_io_file_manager().read(checkpoint_schema_file_path)
+    checkpoint_schema_config = json.loads(schema_file)
 
     if DATAFRAME_PANDERA_SCHEMA_KEY not in checkpoint_schema_config:
         raise ValueError(
@@ -354,7 +357,7 @@ def _get_relative_path(file_path: str) -> str:
         str: The relative path of the file.
 
     """
-    current_directory = os.getcwd()
+    current_directory = get_io_file_manager().getcwd()
     return os.path.relpath(file_path, current_directory)
 
 

snowflake/snowpark_checkpoints/validation_result_metadata.py

@@ -18,6 +18,7 @@ import os
 
 from typing import Optional
 
+from snowflake.snowpark_checkpoints.io_utils.io_file_manager import get_io_file_manager
 from snowflake.snowpark_checkpoints.singleton import Singleton
 from snowflake.snowpark_checkpoints.utils.constants import (
     SNOWPARK_CHECKPOINTS_OUTPUT_DIRECTORY_NAME,
@@ -67,7 +68,9 @@ class ValidationResultsMetadata(metaclass=Singleton):
             Exception: If there is an error reading the validation results file.
 
         """
-        self.validation_results_directory = path if path else os.getcwd()
+        self.validation_results_directory = (
+            path if path else get_io_file_manager().getcwd()
+        )
         self.validation_results_directory = os.path.join(
             self.validation_results_directory,
             SNOWPARK_CHECKPOINTS_OUTPUT_DIRECTORY_NAME,
@@ -89,20 +92,21 @@ class ValidationResultsMetadata(metaclass=Singleton):
 
         self.validation_results = ValidationResults(results=[])
 
-        if os.path.exists(self.validation_results_file):
+        if get_io_file_manager().file_exists(self.validation_results_file):
             LOGGER.info(
                 "Loading validation results from: '%s'", self.validation_results_file
             )
-            with open(self.validation_results_file) as file:
-                try:
-                    validation_result_json = file.read()
-                    self.validation_results = ValidationResults.model_validate_json(
-                        validation_result_json
-                    )
-                except Exception as e:
-                    raise Exception(
-                        f"Error reading validation results file: {self.validation_results_file} \n {e}"
-                    ) from None
+            try:
+                validation_result_json = get_io_file_manager().read(
+                    self.validation_results_file
+                )
+                self.validation_results = ValidationResults.model_validate_json(
+                    validation_result_json
+                )
+            except Exception as e:
+                raise Exception(
+                    f"Error reading validation results file: {self.validation_results_file} \n {e}"
+                ) from None
         else:
             LOGGER.info(
                 "Validation results file not found: '%s'",
@@ -115,7 +119,7 @@ class ValidationResultsMetadata(metaclass=Singleton):
         This method empties the validation results list.
 
         """
-        if not os.path.exists(self.validation_results_file):
+        if not get_io_file_manager().file_exists(self.validation_results_file):
             LOGGER.info("Cleaning validation results...")
             self.validation_results.results = []
 
@@ -123,7 +127,6 @@ class ValidationResultsMetadata(metaclass=Singleton):
         """Add a validation result to the pipeline result list.
 
         Args:
-            checkpoint_name (str): The name of the checkpoint.
             validation_result (dict): The validation result to be added.
 
         """
@@ -140,16 +143,17 @@ class ValidationResultsMetadata(metaclass=Singleton):
             OSError: If the directory cannot be created or the file cannot be written.
 
         """
-        if not os.path.exists(self.validation_results_directory):
+        if not get_io_file_manager().folder_exists(self.validation_results_directory):
             LOGGER.debug(
                 "Validation results directory '%s' does not exist. Creating it...",
                 self.validation_results_directory,
             )
-            os.makedirs(self.validation_results_directory)
+            get_io_file_manager().mkdir(self.validation_results_directory)
 
-        with open(self.validation_results_file, "w") as output_file:
-            output_file.write(self.validation_results.model_dump_json())
-            LOGGER.info(
-                "Validation results successfully saved to: '%s'",
-                self.validation_results_file,
-            )
+        get_io_file_manager().write(
+            self.validation_results_file, self.validation_results.model_dump_json()
+        )
+        LOGGER.info(
+            "Validation results successfully saved to: '%s'",
+            self.validation_results_file,
+        )

{snowpark_checkpoints_validators-0.2.1.dist-info → snowpark_checkpoints_validators-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowpark-checkpoints-validators
-Version: 0.2.1
+Version: 0.3.0
 Summary: Migration tools for Snowpark
 Project-URL: Bug Tracker, https://github.com/snowflakedb/snowpark-checkpoints/issues
 Project-URL: Source code, https://github.com/snowflakedb/snowpark-checkpoints/
@@ -27,9 +27,11 @@ Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Python: <3.12,>=3.9
 Requires-Dist: pandera[io]==0.20.4
+Requires-Dist: pydantic>=2.0
 Requires-Dist: snowflake-connector-python[pandas]
 Requires-Dist: snowflake-snowpark-python>=1.23.0
 Provides-Extra: development
+Requires-Dist: certifi==2025.1.31; extra == 'development'
 Requires-Dist: coverage>=7.6.7; extra == 'development'
 Requires-Dist: deepdiff==8.1.1; extra == 'development'
 Requires-Dist: deepdiff>=8.0.0; extra == 'development'

{snowpark_checkpoints_validators-0.2.1.dist-info → snowpark_checkpoints_validators-0.3.0.dist-info}/RECORD

@@ -1,22 +1,26 @@
 snowflake/snowpark_checkpoints/__init__.py,sha256=p7fzH3f8foD5nhNJHZ00JT3ODTXJGGkWTd3xRKx-8aQ,1435
-snowflake/snowpark_checkpoints/__version__.py,sha256=jEnm4p_P4FqdYsTq3hnGQnhLZ4KwL0_Ew8fDF8BRL98,632
+snowflake/snowpark_checkpoints/__version__.py,sha256=kbbDnlkY7JOLNHvfWYkCO_mOBOV9GniMGdxYoQpLhyg,632
 snowflake/snowpark_checkpoints/checkpoint.py,sha256=i-iDRYbGvQHy9ipW7UxHVhJhQ9BXNSO-bsCcHyg3oLA,22056
 snowflake/snowpark_checkpoints/errors.py,sha256=9KjzRf8bjDZTTNL4LeySJAwuucDOyz0Ka7EFBKWFpyg,1821
 snowflake/snowpark_checkpoints/job_context.py,sha256=RMK0g0HrbDVrOAvai4PgsGvsAn_GIo9aFmh-tWlyieY,4183
 snowflake/snowpark_checkpoints/singleton.py,sha256=7AgIHQBXVRvPBBCkmBplzkdrrm-xVWf_N8svzA2vF8E,836
 snowflake/snowpark_checkpoints/snowpark_sampler.py,sha256=Qxv-8nRGuf-ab3GoSUt8_MNL0ppjoBIMOFIMkqmwN5I,4668
 snowflake/snowpark_checkpoints/spark_migration.py,sha256=s2HqomYx76Hqn71g9TleBeHI3t1nirgfPvkggqQQdts,10253
-snowflake/snowpark_checkpoints/validation_result_metadata.py,sha256=fm2lKxjYlzlL6qsiv2icR9k5o7YNd2OwvFhiqGYrTpo,5745
+snowflake/snowpark_checkpoints/validation_result_metadata.py,sha256=5C8f1g-Grs2ydpXiZBLGt5n9cvEHBaw2-CDeb2vnhpg,5847
 snowflake/snowpark_checkpoints/validation_results.py,sha256=J8OcpNty6hQD8RbAy8xmA0UMbPWfXSmQnHYspWWSisk,1502
+snowflake/snowpark_checkpoints/io_utils/__init__.py,sha256=fmSEYcBGNASBanNvMVW-uv6hcoYre6kEH35K-RliuiA,954
+snowflake/snowpark_checkpoints/io_utils/io_default_strategy.py,sha256=VMfdqj4uDgTEinmpC3D0zXncIB9FxWJod1rI-Yt3YVA,1869
+snowflake/snowpark_checkpoints/io_utils/io_env_strategy.py,sha256=ltG_rxm0CkJFXpskOf__ByZw-C6B9LtycqlyB9EmaJI,3569
+snowflake/snowpark_checkpoints/io_utils/io_file_manager.py,sha256=YHrxRBzTlhIUrSFrsoWkRY_Qa-TXgDWglr00T98Tc5g,2485
 snowflake/snowpark_checkpoints/utils/__init__.py,sha256=I4srmZ8G1q9DU6Suo1S91aVfNvETyisKH95uvLAvEJ0,609
 snowflake/snowpark_checkpoints/utils/constants.py,sha256=pgFttLDQ6fTa6obSdvivWBYClS21ap41YVDNGAS4sxY,4146
-snowflake/snowpark_checkpoints/utils/extra_config.py,sha256=LvOdIhvE450AV0wLVK5P_hANvcNzAv8pLNe7Ksr598U,2802
+snowflake/snowpark_checkpoints/utils/extra_config.py,sha256=xOYaG6MfsUCAHI0C_7qWF_m96xcLIZWwrgxY4UlpaZI,4325
 snowflake/snowpark_checkpoints/utils/logging_utils.py,sha256=yyi6X5DqKeTg0HRhvsH6ymYp2P0wbnyKIzI2RzrQS7k,2278
 snowflake/snowpark_checkpoints/utils/pandera_check_manager.py,sha256=tQIozLO-2kM8WZ-gGKfRwmXBx1cDPaIZB0qIcArp8xA,16100
 snowflake/snowpark_checkpoints/utils/supported_types.py,sha256=GrMX2tHdSFnK7LlPbZx20UufD6Br6TNVRkkBwIxdPy0,1433
-snowflake/snowpark_checkpoints/utils/telemetry.py,sha256=_WOVo19BxcF6cpQDplID6BEOvgJfHTGK1JZI1-OI4uc,31370
-snowflake/snowpark_checkpoints/utils/utils_checks.py,sha256=LFdEzVgirkymXD5LlzuE_lv43yAa3OMIXEnloRAXkGc,14204
-snowpark_checkpoints_validators-0.2.1.dist-info/METADATA,sha256=nhKZaDnpjcwwsH4PTAxqtFCqJEZ_UY-p0J_S5863Tvs,11470
-snowpark_checkpoints_validators-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-snowpark_checkpoints_validators-0.2.1.dist-info/licenses/LICENSE,sha256=pmjhbh6uVhV5MBXOlou_UZgFP7CYVQITkCCdvfcS5lY,11340
-snowpark_checkpoints_validators-0.2.1.dist-info/RECORD,,
+snowflake/snowpark_checkpoints/utils/telemetry.py,sha256=GfuyIaI3QG4a4_qWwyJHvWRM0GENunNexuEJ6IgscF4,32684
+snowflake/snowpark_checkpoints/utils/utils_checks.py,sha256=oQ1c4n-uAA2kFIpWIRPWhbCW8e-wwOIL8qDqLvr5Fok,14398
+snowpark_checkpoints_validators-0.3.0.dist-info/METADATA,sha256=RbOlEHK5kumiBPP2S7-7k7zxzzLYag7Yb6TtQeOYbV0,11557
+snowpark_checkpoints_validators-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+snowpark_checkpoints_validators-0.3.0.dist-info/licenses/LICENSE,sha256=pmjhbh6uVhV5MBXOlou_UZgFP7CYVQITkCCdvfcS5lY,11340
+snowpark_checkpoints_validators-0.3.0.dist-info/RECORD,,