yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-26 16:46:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Support multiple RayPipelines (#4078)

Browse Source
This commit is contained in:
Zoltan Fedor 2023-02-07 05:01:07 -05:00 committed by GitHub
parent 3e4a2201df
commit a3016f065f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
haystack/pipelines/ray.py
View File

@ -77,7 +77,10 @@ class RayPipeline(Pipeline):
:param serve_args: Optional parameters for initializing Ray Serve. :param serve_args: Optional parameters for initializing Ray Serve.
""" """
ray_args = ray_args or {} ray_args = ray_args or {}
if not ray.is_initialized():
ray.init(address=address, **ray_args) ray.init(address=address, **ray_args)
else:
logger.warning("Ray was already initialized, so reusing that for this RayPipeline.")
self._serve_controller_client = serve.start(**serve_args) self._serve_controller_client = serve.start(**serve_args)
super().__init__() super().__init__()