snowpark-checkpoints-collectors 0.2.0__tar.gz → 0.2.1__tar.gz

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowpark-checkpoints-collectors
-Version: 0.2.0
+Version: 0.2.1
 Summary: Snowpark column and table statistics collection
 Project-URL: Bug Tracker, https://github.com/snowflakedb/snowpark-checkpoints/issues
 Project-URL: Source code, https://github.com/snowflakedb/snowpark-checkpoints/

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/src/snowflake/snowpark_checkpoints_collector/__version__.py

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

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/src/snowflake/snowpark_checkpoints_collector/summary_stats_collector.py

@@ -95,7 +95,8 @@ def collect_dataframe_checkpoint(
 
     Raises:
         Exception: Invalid mode value.
-        Exception: Invalid checkpoint name. Checkpoint names must only contain alphanumeric characters and underscores.
+        Exception: Invalid checkpoint name. Checkpoint names must only contain alphanumeric characters
+                   , underscores and dollar signs.
 
     """
     normalized_checkpoint_name = checkpoint_name_utils.normalize_checkpoint_name(
@@ -113,7 +114,7 @@ def collect_dataframe_checkpoint(
     if not is_valid_checkpoint_name:
         raise Exception(
             f"Invalid checkpoint name: {normalized_checkpoint_name}. "
-            f"Checkpoint names must only contain alphanumeric characters and underscores."
+            "Checkpoint names must only contain alphanumeric characters, underscores and dollar signs."
         )
     if not is_checkpoint_enabled(normalized_checkpoint_name):
         LOGGER.info(

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/src/snowflake/snowpark_checkpoints_collector/utils/checkpoint_name_utils.py

@@ -16,7 +16,7 @@
 import re as regx
 
 
-CHECKPOINT_NAME_REGEX_PATTERN = r"[a-zA-Z_][a-zA-Z0-9_]+"
+CHECKPOINT_NAME_REGEX_PATTERN = r"[a-zA-Z_][a-zA-Z0-9_$]+"
 TRANSLATION_TABLE = str.maketrans({" ": "_", "-": "_"})
 
 
@@ -39,7 +39,7 @@ def is_valid_checkpoint_name(checkpoint_name: str) -> bool:
 
     A valid checkpoint name must:
     - Start with a letter (a-z, A-Z) or an underscore (_)
-    - Be followed by any combination of letters, digits (0-9) and underscores (_).
+    - Be followed by any combination of letters, digits (0-9), underscores (_), and dollar signs ($).
 
     Args:
         checkpoint_name (str): The checkpoint name to validate.

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/test/integ/test_checkpoint_name.py

@@ -56,7 +56,7 @@ def test_collect_dataframe_with_invalid_checkpoint_name(
     pyspark_df = spark_session.createDataFrame(data=[], schema=StructType())
     expected_error_msg = (
         f"Invalid checkpoint name: {checkpoint_name}. "
-        f"Checkpoint names must only contain alphanumeric characters and underscores."
+        f"Checkpoint names must only contain alphanumeric characters, underscores and dollar signs."
     )
 
     with (

{snowpark_checkpoints_collectors-0.2.0 → snowpark_checkpoints_collectors-0.2.1}/test/unit/test_checkpoint_name_utils.py

@@ -28,7 +28,8 @@ def test_normalize_checkpoint_name_hyphen_case(input_value, expected_value):
 
 
 @pytest.mark.parametrize(
-    "input_value", ["_checkpoint1", "_checkpoint", "checkPoint1", "Checkpoint", "_1"]
+    "input_value",
+    ["_checkpoint1", "_checkpoint", "checkPoint1", "Checkpoint", "_1", "Dollar$Sample"],
 )
 def test_validate_checkpoint_name_valid_case(input_value):
     is_valid_checkpoint_name = checkpoint_name_utils.is_valid_checkpoint_name(
@@ -38,7 +39,8 @@ def test_validate_checkpoint_name_valid_case(input_value):
 
 
 @pytest.mark.parametrize(
-    "input_value", ["_", "5", "", "56_my_checkpoint", "my-check", "_+check"]
+    "input_value",
+    ["_", "$", "5", "", "56_my_checkpoint", "my-check", "_+check", "$DollarSample"],
 )
 def test_validate_checkpoint_name_invalid_case(input_value):
     is_valid_checkpoint_name = checkpoint_name_utils.is_valid_checkpoint_name(