Skip to content

Commit df2143f

Browse files
author
Payton Staub
committed
Remove old test
1 parent 636f74b commit df2143f

File tree

1 file changed

+0
-26
lines changed

1 file changed

+0
-26
lines changed

tests/unit/sagemaker/workflow/test_steps.py

-26
Original file line numberDiff line numberDiff 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-
526500
def test_processing_step(sagemaker_session):
527501
processing_input_data_uri_parameter = ParameterString(
528502
name="ProcessingInputDataUri", default_value=f"s3://{BUCKET}/processing_manifest"

0 commit comments

Comments
 (0)