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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowpark-checkpoints-configuration
-Version: 0.2.0
+Version: 0.2.1
 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/

{snowpark_checkpoints_configuration-0.2.0 → snowpark_checkpoints_configuration-0.2.1}/src/snowflake/snowpark_checkpoints_configuration/__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_configuration-0.2.0 → snowpark_checkpoints_configuration-0.2.1}/src/snowflake/snowpark_checkpoints_configuration/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_configuration-0.2.0 → snowpark_checkpoints_configuration-0.2.1}/src/snowflake/snowpark_checkpoints_configuration/model/checkpoints.py

@@ -56,7 +56,7 @@ class Checkpoint(BaseModel):
         if not is_valid_checkpoint_name:
             error_msg = (
                 f"Invalid checkpoint name: {name} in checkpoints.json file. "
-                f"Checkpoint names must only contain alphanumeric characters and underscores."
+                f"Checkpoint names must only contain alphanumeric characters, underscores and dollar signs."
             )
             LOGGER.error(error_msg)
             raise Exception(error_msg)

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

@@ -63,7 +63,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):
     checkpoint = Checkpoint(
@@ -79,12 +80,14 @@ def test_validate_checkpoint_name_valid_case(input_value):
     assert checkpoint.name == input_value
 
 
-@pytest.mark.parametrize("input_value", ["_", "5", "", "56_my_checkpoint", "_+check"])
+@pytest.mark.parametrize(
+    "input_value", ["_", "$", "5", "", "56_my_checkpoint", "_+check", "$DollarSample"]
+)
 def test_checkpoint_invalid_name(input_value: str, caplog: pytest.LogCaptureFixture):
     caplog.set_level(level=logging.ERROR, logger=LOGGER_NAME)
     expected_error_msg = (
         f"Invalid checkpoint name: {input_value} in checkpoints.json file. "
-        f"Checkpoint names must only contain alphanumeric characters and underscores."
+        f"Checkpoint names must only contain alphanumeric characters, underscores and dollar signs."
     )
 
     with pytest.raises(Exception) as ex_info: