3d091b43f5c817709299d6cdfb95d4b72b356ca4,src/sagemaker/local/image.py,_SageMakerContainer,_prepare_serving_volumes,#_SageMakerContainer#Any#,300

Before Change

            container_model_dir = os.path.join(self.container_root, host, "model")
            os.makedirs(container_model_dir)

            parsed_uri = urlparse(model_location)
            filename = os.path.basename(parsed_uri.path)
            tar_location = os.path.join(container_model_dir, filename)
            sagemaker.utils.download_file(parsed_uri.netloc, parsed_uri.path, tar_location, self.sagemaker_session)

            if tarfile.is_tarfile(tar_location):
                with tarfile.open(tar_location) as tar:

After Change

        host_dir = os.path.join(self.container_root, host)
        os.makedirs(host_dir)

        model_data_source = sagemaker.local.data.get_data_source_instance(
            model_location, self.sagemaker_session)

        for filename in model_data_source.get_file_list():
            if tarfile.is_tarfile(filename):
                with tarfile.open(filename) as tar:
                    tar.extractall(path=model_data_source.get_root_dir())

        volumes.append(_Volume(model_data_source.get_root_dir(), "/opt/ml/model"))

        return volumes

Instances