Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix keep printing warnings in pipeline (#251)

Browse Source
This commit is contained in:
tastelikefeet
2023-04-10 11:30:40 +08:00
committed by GitHub
parent 27c1bfadde
commit 203a565a39
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modelscope/pipelines/base.py
View File

@@ -339,15 +339,18 @@ class Pipeline(ABC):
check_input_type(input_type[k], input[k])
else:
raise ValueError(f'invalid input_type definition {input_type}')
else:
elif not getattr(self, '_input_has_warned', False):
logger.warning(f'task {task_name} input definition is missing')
self._input_has_warned = True
def _check_output(self, input):
# this attribute is dynamically attached by registry
# when cls is registered in registry using task name
task_name = self.group_key
if task_name not in TASK_OUTPUTS:
logger.warning(f'task {task_name} output keys are missing')
if not getattr(self, '_output_has_warned', False):
logger.warning(f'task {task_name} output keys are missing')
self._output_has_warned = True
return
output_keys = TASK_OUTPUTS[task_name]
missing_keys = []