This will require adding pytest as a dependency. If the plan is to use pip and a requirements.txt for defining dependencies (as opposed to conda), then I would suggest create a separate requirements file for defining development dependencies called requirements-dev.txt and to include pytest in that. This is because it will not be a dependency for running the software (ie training the model), only for running the tests.
Take inspiration from the poetry template test directory.
This will require adding
pytest
as a dependency. If the plan is to usepip
and arequirements.txt
for defining dependencies (as opposed to conda), then I would suggest create a separate requirements file for defining development dependencies calledrequirements-dev.txt
and to includepytest
in that. This is because it will not be a dependency for running the software (ie training the model), only for running the tests.