Patch test, lock mcli version (#990)

mosaicml · Feb 22, 2024 · dff2cf4 · dff2cf4
1 parent 63c88d0
commit dff2cf4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -62,7 +62,7 @@
     'einops==0.7.0',
     'omegaconf>=2.2.3,<3',
     'slack-sdk<4',
-    'mosaicml-cli>=0.5.27,<1',
+    'mosaicml-cli>=0.6.10,<1',
     'onnx==1.14.0',
     'onnxruntime==1.15.1',
     'cmake>=3.25.0,<=3.26.3',  # required for triton-pre-mlir below

diff --git a/tests/callbacks/test_async_eval_callback.py b/tests/callbacks/test_async_eval_callback.py
@@ -231,6 +231,8 @@ def test_validate_eval_run_config():
     cpus=0,
     node_count=2,
     latest_resumption=None,  # type: ignore
+    is_deleted=False,
+    run_type='training',
     submitted_config=RunConfig(
         name=RUN_NAME,
         image='fake-image',