buckification

Differential Revision: D57881544

Pull Request resolved: #918

Author: calebho

dev-requirements.txt

@@ -15,7 +15,7 @@ hydra-core
 ipython
 kfp==1.8.22
 mlflow-skinny
-moto==4.2.14
+moto~=5.0.8
 pyre-extensions
 pyre-check
 pytest

scripts/component_integration_tests.py

@@ -174,13 +174,9 @@ def _mock_aws_batch() -> None:
     ensure_network()
     os.environ.setdefault("MOTO_DOCKER_NETWORK_NAME", NETWORK)
 
-    from moto import mock_batch, mock_ec2, mock_ecs, mock_iam, mock_logs
+    from moto import mock_aws
 
-    mock_batch().__enter__()
-    mock_iam().__enter__()
-    mock_ec2().__enter__()
-    mock_ecs().__enter__()
-    mock_logs().__enter__()
+    mock_aws().__enter__()
 
     import boto3.session