yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2026-01-07 20:46:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add type ignore to resolve mypy errors (#1427)

Browse Source
This commit is contained in:
oryx1729 2021-09-09 12:29:01 +02:00 committed by GitHub
parent a92f1860f6
commit 3e6def7e03
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
haystack/pipeline.py
View File

@ -15,8 +15,8 @@ try:
from ray import serve
import ray
except:
ray = None
serve = None
ray = None # type: ignore
serve = None # type: ignore
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
@ -1136,7 +1136,7 @@ class RayPipeline(Pipeline):
:param replicas: By default, a single replica of the component is created. It can be
configured by setting `replicas` parameter in the Pipeline YAML.
"""
RayDeployment = serve.deployment(_RayDeploymentWrapper, name=component_name, num_replicas=replicas)
RayDeployment = serve.deployment(_RayDeploymentWrapper, name=component_name, num_replicas=replicas) # type: ignore
RayDeployment.deploy(pipeline_config, component_name)
handle = RayDeployment.get_handle()
return handle