Source code for nboost.compat

from nboost.maps import CLASS_MAP, URL_MAP, MODULE_MAP


[docs]class BackwardsCompatibility: """Augment global modules to be backwards compatible"""
[docs] def set(self): MODULE_MAP['QAModel'] = MODULE_MAP['QAModelPlugin'] MODULE_MAP['RerankModel'] = MODULE_MAP['RerankModelPlugin'] MODULE_MAP['ShuffleModel'] = MODULE_MAP['ShuffleRerankModelPlugin'] MODULE_MAP['PtBertModel'] = MODULE_MAP['PtBertRerankModelPlugin'] MODULE_MAP['TfBertModel'] = MODULE_MAP['TfBertRerankModelPlugin'] MODULE_MAP['TfAlbertModel'] = MODULE_MAP['TfAlbertRerankModelPlugin'] MODULE_MAP['PtDistilBertQAModel'] = MODULE_MAP['PtDistilBertQAModelPlugin'] URL_MAP["bert-base-uncased-msmarco"] = URL_MAP["tf-bert-base-uncased-msmarco"] URL_MAP["albert-tiny-uncased-msmarco"] = URL_MAP["tf-albert-tiny-uncased-msmarco"] URL_MAP["biobert-base-uncased-msmarco"] = URL_MAP["tf-biobert-base-uncased-msmarco"] CLASS_MAP["bert-base-uncased-msmarco"] = CLASS_MAP["tf-bert-base-uncased-msmarco"] CLASS_MAP["albert-tiny-uncased-msmarco"] = CLASS_MAP["tf-albert-tiny-uncased-msmarco"] CLASS_MAP["biobert-base-uncased-msmarco"] = CLASS_MAP["tf-biobert-base-uncased-msmarco"]