aws / sagemaker-pytorch-inference-toolkit

Toolkit for allowing inference and serving with PyTorch on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at https://github.com/aws/deep-learning-containers.
Apache License 2.0
134 stars 72 forks source link

reuse sagemaker-inference's requirements.txt installation logic #150

Closed humanzz closed 1 year ago

humanzz commented 1 year ago

*Issue #, if available: #149

Description of changes:

use the REQUIREMENTS_PATH and _install_requirements() from sagemaker-inference-toolkit to get CodeArtifact support for free from https://github.com/aws/sagemaker-inference-toolkit/pull/130 and to reduce code duplication

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.