Merge pull request aws#4 from aws/test_fixes

Test fixes

Author: ertukara0

README.rst

@@ -1,4 +1,4 @@
-|codebuild|
+|codebuild| |readthedocs| |pypi|
 
 ===================================
 AWS Step Functions Data Science SDK
@@ -559,5 +559,18 @@ The following are details about the warnings you might see:
 
 For more information, see http://www.gnupg.org.
 
+.. |codebuild| image:: https://codebuild.us-east-2.amazonaws.com/badges?uuid=eyJlbmNyeXB0ZWREYXRhIjoiZ2crZkxWN2lPTHhBdzAwOUIvZDlUQ2txQTRyYnZnQ3RaQ0dQYkhsb2EvT04xOVRIdDBqYWFOaS8weklGU216OUtuc29pZFQvQjgrRDhRbWJoeEJocFV3PSIsIml2UGFyYW1ldGVyU3BlYyI6IlRQUlZQd1ZLdGRqWkdVdWkiLCJtYXRlcmlhbFNldFNlcmlhbCI6MX0%3D&branch=master
+ :target: https://us-east-2.console.aws.amazon.com/codesuite/codebuild/projects/StepFunctionsPythonSDK-unittests-private/history?region=us-east-2
+ :alt: Unit Tests Build Status
 
-.. |codebuild| image:: https://codebuild.us-east-2.amazonaws.com/badges?uuid=eyJlbmNyeXB0ZWREYXRhIjoiUkFzRXd6UmdKZkJIZFRPMTRCMmhKYzJqL1U0bEpMdDFvSGJPeXBCSlhQaDBaQVZxYWtnUkZNMmhlclRSeGxCbjZhVTl0dlpiQXFKd1puUFZJK0xmNHN3PSIsIml2UGFyYW1ldGVyU3BlYyI6ImZ2ekJpa3V5ZXgxV3gyczUiLCJtYXRlcmlhbFNldFNlcmlhbCI6MX0%3D&branch=master
+.. |codebuild| image:: https://codebuild.us-east-2.amazonaws.com/badges?uuid=eyJlbmNyeXB0ZWREYXRhIjoiZ2crZkxWN2lPTHhBdzAwOUIvZDlUQ2txQTRyYnZnQ3RaQ0dQYkhsb2EvT04xOVRIdDBqYWFOaS8weklGU216OUtuc29pZFQvQjgrRDhRbWJoeEJocFV3PSIsIml2UGFyYW1ldGVyU3BlYyI6IlRQUlZQd1ZLdGRqWkdVdWkiLCJtYXRlcmlhbFNldFNlcmlhbCI6MX0%3D&branch=master
+ :target: https://us-east-2.console.aws.amazon.com/codesuite/codebuild/projects/StepFunctionsPythonSDK-unittests-private/history?region=us-east-2
+ :alt: Unit Tests Build Status
+
+.. |readthedocs| image:: https://readthedocs.org/projects/aws-step-functions-data-science-sdk/badge/?version=latest
+ :target: https://aws-step-functions-data-science-sdk.readthedocs.io/en/latest/?badge=latest
+ :alt: Documentation Status
+
+.. |pypi| image:: https://img.shields.io/pypi/v/stepfunctions
+ :target: https://pypi.org/project/stepfunctions/
+ :alt: PyPI

setup.py

@@ -67,7 +67,7 @@ def read_version():
  "tensorflow>=1.3.0",
  "mock>=2.0.0",
  "contextlib2==0.5.5",
- "IPython>=7.8.0"
+ "IPython"
  ]
  }
 )

tests/integ/test_inference_pipeline.py

@@ -123,6 +123,6 @@ def test_inference_pipeline_framework(
  assert client_info['input'] == json.loads(execution_info['input'])
 
  state_machine_delete_wait(sfn_client, state_machine_arn)
- delete_sagemaker_model(job_name, sagemaker_session)
- delete_sagemaker_endpoint_config(job_name, sagemaker_session)
  delete_sagemaker_endpoint(job_name, sagemaker_session)
+ delete_sagemaker_endpoint_config(job_name, sagemaker_session)
+ delete_sagemaker_model(job_name, sagemaker_session)

tests/integ/test_sagemaker_steps.py

@@ -203,8 +203,8 @@ def test_endpoint_config_step(trained_estimator, sfn_client, sagemaker_session,
 
  # Cleanup
  state_machine_delete_wait(sfn_client, workflow.state_machine_arn)
- delete_sagemaker_model(model.name, sagemaker_session)
  delete_sagemaker_endpoint_config(endpoint_config_name, sagemaker_session)
+ delete_sagemaker_model(model.name, sagemaker_session)
  # End of Cleanup
 
 def test_create_endpoint_step(trained_estimator, record_set_fixture, sfn_client, sagemaker_session, sfn_role_arn):
@@ -244,9 +244,9 @@ def test_create_endpoint_step(trained_estimator, record_set_fixture, sfn_client,
 
  # Cleanup
  state_machine_delete_wait(sfn_client, workflow.state_machine_arn)
- delete_sagemaker_model(model.name, sagemaker_session)
- delete_sagemaker_endpoint_config(model.name, sagemaker_session)
  delete_sagemaker_endpoint(endpoint_name, sagemaker_session)
+ delete_sagemaker_endpoint_config(model.name, sagemaker_session)
+ delete_sagemaker_model(model.name, sagemaker_session)
  # End of Cleanup
 
 def test_tuning_step(sfn_client, record_set_for_hyperparameter_tuning, sagemaker_role_arn, sfn_role_arn):

tests/integ/test_training_pipeline_estimators.py

@@ -151,6 +151,6 @@ def test_pca_estimator(sfn_client, sagemaker_session, sagemaker_role_arn, sfn_ro
 
  # Cleanup
  state_machine_delete_wait(sfn_client, state_machine_arn)
- delete_sagemaker_model(job_name, sagemaker_session)
- delete_sagemaker_endpoint_config(job_name, sagemaker_session)
  delete_sagemaker_endpoint(job_name, sagemaker_session)
+ delete_sagemaker_endpoint_config(job_name, sagemaker_session)
+ delete_sagemaker_model(job_name, sagemaker_session)