## Local development

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

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

## Tests

### Tests against an Amazon SageMaker Distribution image

The [test_dockerfile_based_harness.py](test/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.

Run the following to invoke those tests:

```shell
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.