Skip to content

Commit 1bb489a

Browse files
authored
Merge pull request #597 from microsoft/staging
Sync master <-> staging
2 parents 6608a90 + 1e8e6ad commit 1bb489a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/.ci/submit_azureml_pytest.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from azureml.core.compute import ComputeTarget, AmlCompute
4747
from azureml.core.compute_target import ComputeTargetException
4848
from azureml.core.workspace import WorkspaceException
49+
import azureml._restclient.snapshots_client
4950

5051

5152
def setup_workspace(
@@ -384,6 +385,9 @@ def create_arg_parser():
384385
logger.setLevel(logging.INFO)
385386
# logging.basicConfig(level=logging.DEBUG)
386387
args = create_arg_parser()
388+
389+
# Allow snapshots of more than 300MB
390+
azureml._restclient.snapshots_client. SNAPSHOT_MAX_SIZE_BYTES = 1000*(1024*1024)
387391

388392
if args.dockerproc == "cpu":
389393
from azureml.core.runconfig import DEFAULT_CPU_IMAGE

0 commit comments

Comments
 (0)