def test_download_file_no_collision(self, experiment_run, dir_and_files,
in_tempdir):
source_dirpath, _ = dir_and_files
key = "artifact"
# create archive and move into cwd so it's deleted on teardown
filepath = os.path.abspath("archive.zip")
temp_zip = _artifact_utils.zip_dir(source_dirpath)
os.rename(temp_zip.name, filepath)
# upload and download file
experiment_run.log_artifact(key, filepath)
download_url = experiment_run._get_url_for_artifact(key, "GET").url
response = requests.get(download_url)
downloaded_filepath = _request_utils.download_file(
response,
filepath,
overwrite_ok=False,
)
downloaded_filepath = os.path.abspath(downloaded_filepath)
# different names
assert filepath != downloaded_filepath
# contents match
assert filecmp.cmp(filepath, downloaded_filepath)
def download_docker_context(self, download_to_path, self_contained=False):
"""
Downloads this Model Version's Docker context ``tgz``.
Parameters
----------
download_to_path : str
Path to download Docker context to.
self_contained : bool, default False
Whether the downloaded Docker context should be self-contained.
Returns
-------
downloaded_to_path : str
Absolute path where Docker context was downloaded to. Matches `download_to_path`.
"""
self._refresh_cache()
endpoint = "{}://{}/api/v1/deployment/builds/dockercontext".format(
self._conn.scheme,
self._conn.socket,
)
body = {
"model_version_id": self.id,
"self_contained": self_contained,
}
with _utils.make_request("POST",
endpoint,
self._conn,
json=body,
stream=True) as response:
try:
_utils.raise_for_http_error(response)
except requests.HTTPError as e:
# propagate error caused by missing artifact
error_text = e.response.text.strip()
if error_text.startswith("missing artifact"):
new_e = RuntimeError(
"unable to obtain Docker context due to " + error_text)
six.raise_from(new_e, None)
else:
raise e
downloaded_to_path = _request_utils.download_file(
response, download_to_path, overwrite_ok=True)
return os.path.abspath(downloaded_to_path)