buckification (#918)

Summary: This commit was generated using `mgt import`. buckification for third-party libraries: third-party/pypi/responses/0.25.0 third-party/pypi/moto/5.0.8 uuid_e3b367cdbe054c858e46ee4fad2a911b Differential Revision: D57881544
pytorch · May 29, 2024 · 1cfaad0 · 1cfaad0
1 parent a2423f4
commit 1cfaad0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/dev-requirements.txt b/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

diff --git a/scripts/component_integration_tests.py b/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