Browse Source

[to #42322933] fix modelscope/pipelines/base.py

删除base.py _collate_fn()函数中关于text2sql模型的相关代码,挪到ConversationalTextToSqlPipeline相关的代码里
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9730687
master
piaoyu.lxy yingda.chen 3 years ago
parent
commit
be6b82cd6d
2 changed files with 3 additions and 3 deletions
  1. +0
    -3
      modelscope/pipelines/base.py
  2. +3
    -0
      modelscope/pipelines/nlp/conversational_text_to_sql_pipeline.py

+ 0
- 3
modelscope/pipelines/base.py View File

@@ -239,7 +239,6 @@ class Pipeline(ABC):
"""
from torch.utils.data.dataloader import default_collate
from modelscope.preprocessors import InputFeatures
from text2sql_lgesql.utils.batch import Batch
if isinstance(data, dict) or isinstance(data, Mapping):
return type(data)(
{k: self._collate_fn(v)
@@ -260,8 +259,6 @@ class Pipeline(ABC):
return data
elif isinstance(data, InputFeatures):
return data
elif isinstance(data, Batch):
return data
else:
import mmcv
if isinstance(data, mmcv.parallel.data_container.DataContainer):


+ 3
- 0
modelscope/pipelines/nlp/conversational_text_to_sql_pipeline.py View File

@@ -64,3 +64,6 @@ class ConversationalTextToSqlPipeline(Pipeline):
sql = Example.evaluator.obtain_sql(inputs['predict'][0], inputs['db'])
result = {OutputKeys.TEXT: sql}
return result

def _collate_fn(self, data):
return data

Loading…
Cancel
Save