Python BentoServiceEnv.add_pip_packages Examples

Example #1

 def set_dependencies(self, env: BentoServiceEnv):
     # Note that keras module is not required, user can use tf.keras as an
     # replacement for the keras module. Although tensorflow module is required to
     #  be used as the default Keras backend
     pip_deps = ['tensorflow']
     if self._keras_module_name == 'keras':
         pip_deps.append('keras')
     env.add_pip_packages(pip_deps)

Example #2

 def set_dependencies(self, env: BentoServiceEnv):
     logger.warning(
         "BentoML by default does not include spacy and torchvision package when "
         "using FastaiModelArtifact. To make sure BentoML bundle those packages if "
         "they are required for your model, either import those packages in "
         "BentoService definition file or manually add them via "
         "`@env(pip_packages=['torchvision'])` when defining a BentoService"
     )
     env.add_pip_packages(['torch', "fastai<2.0.0"])

Example #3

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['coremltools>=4.0b2'])

Example #4

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(["fasttext"])

Example #5

File: onnx.py Project: Nikunj3masarani/bentoml

 def set_dependencies(self, env: BentoServiceEnv):
     if self.backend == "onnxruntime":
         env.add_pip_packages(["onnxruntime"])
     elif self.backend == "onnxruntime-gpu":
         env.add_pip_packages(["onnxruntime-gpu"])

Example #6

File: tensorflow.py Project: Nikunj3masarani/bentoml

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['tensorflow'])

Example #7

File: xgboost.py Project: subhayuroy/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     if env._infer_pip_packages:
         env.add_pip_packages(['xgboost'])

Example #8

File: spacy.py Project: zhangjiashen/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['spacy'])

Example #9

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['torch', "detectron2"])

Example #10

File: easyocr.py Project: snsnlou/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(["easyocr>=1.3.0"])

Example #11

File: h2o.py Project: zhangjiashen/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['h2o'])
     env.add_conda_dependencies(['openjdk'])

Example #12

File: gluon.py Project: subhayuroy/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     if env._infer_pip_packages:
         env.add_pip_packages(["mxnet"])

Example #13

File: onnxmlir.py Project: snsnlou/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(["numpy"])

Example #14

File: pytorch.py Project: Nikunj3masarani/bentoml

 def set_dependencies(self, env: BentoServiceEnv):
     env.add_pip_packages(['pytorch-lightning'])

Example #15

File: onnx.py Project: joerg84/BentoML

 def set_dependencies(self, env: BentoServiceEnv):
     if self.backend == 'onnxruntime':
         env.add_pip_packages(['onnxruntime'])

Example #16

 def set_dependencies(self, env: BentoServiceEnv):
     if env._infer_pip_packages:
         env.add_pip_packages(['scikit-learn'])