in src/deep_learning_container.py [0:0]
def query_bucket(instance_id, region):
"""
GET request on an empty object from an Amazon S3 bucket
"""
response = None
args = parse_args()
framework, framework_version, container_type = (
args.framework,
args.framework_version,
args.container_type,
)
py_version = sys.version.split(" ")[0]
if instance_id is not None and region is not None:
url = (
"https://aws-deep-learning-containers-{0}.s3.{1}.amazonaws.com"
"/dlc-containers-{2}.txt?x-instance-id={2}&x-framework={3}&x-framework_version={4}&x-py_version={5}&x-container_type={6}".format(
REGION_MAPPING[region],
region,
instance_id,
framework,
framework_version,
py_version,
container_type,
)
)
response = requests_helper(url, timeout=0.2)
if os.environ.get("TEST_MODE") == str(1):
with open(os.path.join(os.sep, "tmp", "test_request.txt"), "w+") as rf:
rf.write(url)
logging.debug("Query bucket finished: {}".format(response))
return response