From c6f7b4c6cd5c78283e21d232a652af1e32cfa2c6 Mon Sep 17 00:00:00 2001 From: Daniel King Date: Fri, 14 Jun 2024 15:31:07 -0700 Subject: [PATCH 1/2] add dep group --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index f81b1cd0f1..fd746f546f 100644 --- a/setup.py +++ b/setup.py @@ -54,7 +54,7 @@ ] install_requires = [ - 'mosaicml[libcloud,wandb,oci,gcs]>=0.23.2,<0.24', + 'mosaicml[libcloud,wandb,oci,gcs,mlflow]>=0.23.2,<0.24', 'mlflow>=2.12.1,<2.13', 'accelerate>=0.25,<0.26', # for HF inference `device_map` 'transformers>=4.40,<4.41', From 356dbb59d5f6ebe88edc8e50443167208079862d Mon Sep 17 00:00:00 2001 From: Daniel King Date: Fri, 14 Jun 2024 16:08:46 -0700 Subject: [PATCH 2/2] pc --- scripts/data_prep/convert_delta_to_json.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/data_prep/convert_delta_to_json.py b/scripts/data_prep/convert_delta_to_json.py index d871761803..f63f1b0027 100644 --- a/scripts/data_prep/convert_delta_to_json.py +++ b/scripts/data_prep/convert_delta_to_json.py @@ -463,6 +463,7 @@ def validate_and_get_cluster_info( if res is None: raise ClusterDoesNotExistError(cluster_id) + assert res.spark_version is not None stripped_runtime = re.sub( r'[a-zA-Z]', '',