third_codes/eland
1
0
Fork 0
mirror of https://github.com/elastic/eland.git synced 2025-07-11 00:02:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[NLP] Prevent TypeError with None check (#525)

Browse Source
This commit is contained in:
David Kyle 2023-04-03 14:56:19 +01:00 committed by GitHub
parent cebee6406f
commit 8e0d897171
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
eland/ml/pytorch/transformers.py
View File

@ -393,11 +393,13 @@ class _DPREncoderWrapper(nn.Module): # type: ignore
def is_compatible() -> bool: def is_compatible() -> bool:
is_dpr_model = config.model_type == "dpr" is_dpr_model = config.model_type == "dpr"
has_architectures = len(config.architectures) == 1 has_architectures = (
is_supported_architecture = ( config.architectures is not None and len(config.architectures) == 1
)
is_supported_architecture = has_architectures and (
config.architectures[0] in _DPREncoderWrapper._SUPPORTED_MODELS_NAMES config.architectures[0] in _DPREncoderWrapper._SUPPORTED_MODELS_NAMES
) )
return is_dpr_model and has_architectures and is_supported_architecture return is_dpr_model and is_supported_architecture
if is_compatible(): if is_compatible():
model = getattr(transformers, config.architectures[0]).from_pretrained( model = getattr(transformers, config.architectures[0]).from_pretrained(