File tree 1 file changed +0
-26
lines changed
tests/unit/sagemaker/workflow
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):
497
497
assert len (w ) == 0
498
498
499
499
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
-
526
500
def test_processing_step (sagemaker_session ):
527
501
processing_input_data_uri_parameter = ParameterString (
528
502
name = "ProcessingInputDataUri" , default_value = f"s3://{ BUCKET } /processing_manifest"
You can’t perform that action at this time.
0 commit comments