Make sure to have Git and Python3.7+ installed.
Optionally, make sure to have GNU Make available on your system if you aren't using UNIX-based system for better developer experience.
If you don't want to use make
then refers to Makefile for specific commands on given make target.
python --version
pip --version
Clone the source code from BentoML's GitHub repository:
git clone https://github.com/bentoml/BentoML.git && cd BentoML
Install BentoML with pip in editable
mode:
pip install -e .
This will make bentoml
available on your system which links to the sources of
your local clone and pick up changes you made locally.
Test the BentoML installation either with bash
or in an IPython session:
bentoml --version
print(bentoml.__version__)
pip
also supports installing directly from remote git repository. This makes it
easy to try out new BentoML feature that has not been released, test changes in a pull
request. For example, to install BentoML from its main branch:
pip install git+https://github.com/bentoml/BentoML.git
Or to install from your own fork of BentoML:
pip install git+https://github.com/{your_github_username}/BentoML.git
You can also specify what branch to install from:
pip install git+https://github.com/{your_github_username}/BentoML.git@{branch_name}
It is possible to force a Bento to use a custom BentoML source distribution:
- Install custom BentoML in editable mode. e.g.:
- git clone your bentoml fork
pip install -e PATH_TO_THE_FORK
- Set env var
export BENTOML_BUNDLE_LOCAL_BUILD=True
- make sure you have the latest setuptools installed:
pip install -U setuptools
- make sure you have the latest setuptools installed:
- Build a new Bento with
bentoml build
in your project directory - The new Bento will include a wheel file built from the BentoML source, and
bentoml containrize
will install it to override the default BentoML installation in base image
Make sure to install all test dependencies:
pip install -r requirements/tests-requirements.txt
If you are adding new frameworks it is recommended that you also added tests for our CI. Currently we are using GitHub Actions to manage our CI/CD workflow.
We recommended you to use nektos/act
to run and tests Actions locally.
We introduce a tests script run_tests.sh that can be used to run tests locally and on CI.
./scripts/ci/run_tests.sh -h
Running unit/integration tests with pytest and generate coverage reports. Make sure that given testcases is defined under ./scripts/ci/config.yml.
Usage:
./scripts/ci/run_tests.sh [-h|--help] [-v|--verbose] <target> <pytest_additional_arguments>
Flags:
-h, --help show this message
-v, --verbose set verbose scripts
If pytest_additional_arguments is given, this will be appended to given tests run.
Example:
$ ./scripts/ci/run_tests.sh pytorch --gpus --capture=tee-sys
All tests are then defined under config.yml where each fields follow the following format:
<target>: &tmpl
root_test_dir: "tests/integration/frameworks"
is_dir: false
override_name_or_path:
dependencies: []
external_scripts:
type_tests: "integration"
By default, each of our frameworks tests file with have the format: test_<frameworks>_impl.py
. If is_dir
set to true
we will try to match the given <target>
under root_test_dir
to run tests from.
Keys | Type | Defintions |
---|---|---|
root_test_dir |
<str> |
root directory to run a given tests |
is_dir |
<bool> |
whether target is a directory instead of a file |
override_name_or_path |
<str> |
optional way to override a tests file name if doesn't match our convention |
dependencies |
<List[str]> |
define additional dependencies required to run the tests, accepts requirements.txt format |
external_scripts |
<str> |
optional shell scripts that can be run on top of ./scripts/ci/run_tests.sh for given testsuite |
type_tests |
<Literal["e2e","unit","integration"]> |
define type of tests for given target |
When type_tests
is set to e2e
, ./scripts/ci/run_tests.sh
will change current directory into given root_test_dir
and will run testsuite from there.
The reason why we encourage developers to use the scripts in CI is that under the hood when we uses pytest, we will create a custom reports for given tests. This report then can be used as carryforward flags on codecov for consistent reporting.
Example:
# e2e tests
general_features:
root_test_dir: "tests/e2e/bento_server_general_features"
is_dir: true
type_tests: "e2e"
dependencies:
- "Pillow"
# framework
pytorch_lightning:
<<: *tmpl
dependencies:
- "pytorch-lightning"
- "-f https://download.pytorch.org/whl/torch_stable.html"
- "torch==1.9.0+cpu"
- "torchvision==0.10.0+cpu"
Refers to config.yml for more examples.
You can do this in two ways:
Run all unit tests directly with pytest:
# GIT_ROOT=$(git rev-parse --show-toplevel)
pytest tests/unit --cov=bentoml --cov-config="$GIT_ROOT"/setup.cfg
Run all unit tests via ./scripts/ci/run_tests.sh
:
./scripts/ci/run_tests.sh unit
# on UNIX-based system
make tests-unit
Run given tests after defining target under scripts/ci/config.yml
with run_tests.sh
:
# example: run Keras TF1 integration tests
./scripts/ci/run_tests.sh keras_tf1
# example: run e2e tests to check for general features
./scripts/ci/run_tests.sh general_features
Add the following lines to the Python code that invokes BentoML:
from bentoml.configuration import set_debug_mode
set_debug_mode(True)
And/or set the BENTOML_DEBUG
environment variable to TRUE
:
export BENTOML_DEBUG=TRUE
And/or use the --verbose
option when running bentoml
CLI command, e.g.:
bentoml get IrisClassifier --verbose
linter: pylint
type checker: pyright
Run linter/format script:
make format
make lint
Run type checker:
make type
Refers to BentoML Documentation for more information
Install all docs dependencies:
pip install -r requirements/docs-requirements.txt
To build documentation for locally:
cd docs/
make clean && make html
Modify *.rst
files inside the docs
folder to update content, and to
view your changes, run the following command:
python -m http.server --directory ./docs/build/html
Docs can then be accessed at localhost:8000
If you are developing under macOS or Linux, we also made a script that watches docs file changes, automatically rebuild the docs, and refreshes the browser tab to show the change (UNIX-based system only):
./scripts/watch_docs.sh
Install spellchecker dependencies:
make install-spellchecker-deps
To run spellchecker locally:
make spellcheck-doc
Make sure you have fswatch command installed:
brew install fswatch
Make sure you have inotifywait
installed
sudo apt install inotify-tools
Currently BentoML are PEP518 compatible via setup.cfg
and pyproject.toml
.
We also define most of our config for Python tools where:
pyproject.toml
contains config forsetuptools
,black
,pytest
,pylint
,isort
,pyright
setup.cfg
contains metadata forbentoml
library andcoverage
BentoML has moved its benchmark to bentoml/benchmark
.
BentoML also provides git hooks that developers can install with:
make hooks
Refers to Installation to install pyright correctly.
In order to make pyright function correctly one also need to run the following scripts alongside with the stubs provided in the main repository.
One can also clone a copy of all dependencies stubs used by BentoML to typings/
via:
# Assuming at $GIT_ROOT
git clone [email protected]:bentoml/stubs.git
\rm -rf typings/.git* typings/*.{sh,toml,txt,cfg,py,md}
Fork BentoML project on GitHub and add upstream remotes to local BentoML clone:
git remote add upstream [email protected]:bentoml/BentoML.git
Make the changes either to fix a known issue or adding new feature
In order for us to manage PR and Issues systematically, we encourage developers to use hierarchical branch folders to manage branch naming.
Run ./dev/install_git_hooks.sh
to install pre-commit
hooks. We will check if your branch naming
follows the given regex : ^(feature|bugfix|improv|lib|prerelease|release|hotfix)\/[a-zA-Z0-9._-]+$
. This
is partially based on how Azure DevOps
manages its repos and conventional commits spec (see below for more information):
- feature: new features/proposals users want to integrate into the library, not a new feature for a build script
- bugfix: bugfix of a feature, not a fix to a build script
- improv: improvements/refactor/cleanup production code, eg. reformat, pylint, etc.
- lib: related to internal libraries, features required by the production code
- prerelease: alpha/beta features that might should be included in the prerelease of the library. This would help testing new features/integrations for the library
- release: included all features that is production ready
- hotifx: patch of current bugs in production code
Push changes to your fork and follow this article on how to create a pull request on github. Name your pull request with one of the following prefixes, e.g. "feat: add support for PyTorch". This is based on the Conventional Commits specification
- feat: (new feature for the user, not a new feature for build script)
- fix: (bug fix for the user, not a fix to a build script)
- docs: (changes to the documentation)
- style: (formatting, missing semicolons, etc; no production code change)
- refactor: (refactoring production code, eg. renaming a variable)
- perf: (code changes that improve performance)
- test: (adding missing tests, refactoring tests; no production code change)
- chore: (updating grunt tasks etc; no production code change)
- build: (changes that affect the build system or external dependencies)
- ci: (changes to configuration files and scripts)
- revert: (reverts a previous commit)
Once your pull request created, an automated test run will be triggered on your branch and the BentoML authors will be notified to review your code changes. Once tests are passed and reviewer has signed off, we will merge your pull request.