Skip to content

Latest commit

 

History

History
31 lines (20 loc) · 895 Bytes

DEVELOPMENT.md

File metadata and controls

31 lines (20 loc) · 895 Bytes
Raw

Local development

This project uses Conda to manage its dependencies. Run the following to setup your local environment:

conda env update --file environment.yml -n sagemaker-distribution

Tests

Tests against an Amazon SageMaker Distribution image

The test_dockerfile_based_harness.py contains test cases that can be run against an Amazon SageMaker Distribution image: this could be a pre-built image from our ECR repository, or it could be an image that you built locally using the build command:

python main.py build --target-patch-version x.y.z

Run the following to invoke those tests:

pytest --local-image-id REPLACE_ME_WITH_IMAGE_ID

You can also pass a --use-gpu flag if the test machine has Nvidia GPU(s) and necessary Nvidia drivers.

Unit tests for the project's source code

TODO.