예제 #1
0
파일: service.py 프로젝트: 1e0ng/BentoML
    def _init_env(self, env=None):
        if env is None:
            # By default use BentoServiceEnv defined on class via @env decorator
            env = self.__class__._env

        if isinstance(env, dict):
            self._env = BentoServiceEnv.from_dict(env)
        else:
            self._env = env
예제 #2
0
    def _init_env(self, env=None):
        if env is None:
            # By default use BentoServiceEnv defined on class via @env decorator
            env = self.__class__._env

        if isinstance(env, dict):
            self._env = BentoServiceEnv.from_dict(env)
        else:
            self._env = env

        for api in self._service_apis:
            self._env.add_handler_dependencies(api.handler.pip_dependencies)
예제 #3
0
 def decorator(bento_service_cls):
     bento_service_cls._env = BentoServiceEnv.from_dict(kwargs)
     return bento_service_cls