fix regex (#877)

mosaicml · Jan 17, 2024 · f43d1cf · f43d1cf
1 parent b40b805
commit f43d1cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/data_prep/convert_delta_to_json.py b/scripts/data_prep/convert_delta_to_json.py
@@ -407,7 +407,7 @@ def validate_and_get_cluster_info(cluster_id: str,
         stripped_runtime = re.sub(
             r'[a-zA-Z]', '',
             res.spark_version.split('-scala')[0].replace('x-snapshot', ''))
-        runtime_version = re.sub(r'.-+$', '', stripped_runtime)
+        runtime_version = re.sub(r'[.-]*$', '', stripped_runtime)
         if version.parse(runtime_version) < version.parse(
                 MINIMUM_SQ_CONNECT_DBR_VERSION):
             raise ValueError(