File tree Expand file tree Collapse file tree 1 file changed +0
-26
lines changed
tests/unit/sagemaker/workflow Expand file tree Collapse file tree 1 file changed +0
-26
lines changed Original file line number Diff line number Diff line change @@ -497,32 +497,6 @@ def test_training_step_no_profiler_warning(sagemaker_session):
497497 assert len (w ) == 0
498498
499499
500- def test_training_step_profiler_not_explicitly_enabled (sagemaker_session ):
501- estimator = TensorFlow (
502- entry_point = DUMMY_SCRIPT_PATH ,
503- role = ROLE ,
504- model_dir = False ,
505- image_uri = IMAGE_URI ,
506- source_dir = "s3://mybucket/source" ,
507- framework_version = "2.4.1" ,
508- py_version = "py37" ,
509- instance_count = 1 ,
510- instance_type = "ml.p3.16xlarge" ,
511- sagemaker_session = sagemaker_session ,
512- hyperparameters = {
513- "batch-size" : 500 ,
514- "epochs" : 5 ,
515- },
516- debugger_hook_config = False ,
517- distribution = {"smdistributed" : {"dataparallel" : {"enabled" : True }}},
518- )
519-
520- inputs = TrainingInput (s3_data = f"s3://{ BUCKET } /train_manifest" )
521- step = TrainingStep (name = "MyTrainingStep" , estimator = estimator , inputs = inputs )
522- step_request = step .to_request ()
523- assert step_request ["Arguments" ]["ProfilerRuleConfigurations" ] is None
524-
525-
526500def test_processing_step (sagemaker_session ):
527501 processing_input_data_uri_parameter = ParameterString (
528502 name = "ProcessingInputDataUri" , default_value = f"s3://{ BUCKET } /processing_manifest"
You can’t perform that action at this time.
0 commit comments