From f45d28cb4a36cab1bd22747ed0d5dfda8bd5397b Mon Sep 17 00:00:00 2001
From: Caleb Ho <calebh@meta.com>
Date: Thu, 30 May 2024 10:40:34 -0700
Subject: [PATCH] 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
---
 dev-requirements.txt                   | 2 +-
 scripts/component_integration_tests.py | 8 ++------
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/dev-requirements.txt b/dev-requirements.txt
index bee78d4cc..8dbcac2ec 100644
--- 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
index c4a512c3d..2eb3013d4 100755
--- 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