55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
# Code generated by Lark OpenAPI.
|
||
|
|
|
||
|
|
from typing import Any, Optional, Union, Dict, List, Set, IO, Callable, Type
|
||
|
|
from lark_oapi.core.construct import init
|
||
|
|
|
||
|
|
|
||
|
|
class ModelParam(object):
|
||
|
|
_types = {
|
||
|
|
"encoder_name": str,
|
||
|
|
"ranker_name": str,
|
||
|
|
"filter_name": str,
|
||
|
|
"booster_name": str,
|
||
|
|
"passage_language": str,
|
||
|
|
}
|
||
|
|
|
||
|
|
def __init__(self, d=None):
|
||
|
|
self.encoder_name: Optional[str] = None
|
||
|
|
self.ranker_name: Optional[str] = None
|
||
|
|
self.filter_name: Optional[str] = None
|
||
|
|
self.booster_name: Optional[str] = None
|
||
|
|
self.passage_language: Optional[str] = None
|
||
|
|
init(self, d, self._types)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def builder() -> "ModelParamBuilder":
|
||
|
|
return ModelParamBuilder()
|
||
|
|
|
||
|
|
|
||
|
|
class ModelParamBuilder(object):
|
||
|
|
def __init__(self) -> None:
|
||
|
|
self._model_param = ModelParam()
|
||
|
|
|
||
|
|
def encoder_name(self, encoder_name: str) -> "ModelParamBuilder":
|
||
|
|
self._model_param.encoder_name = encoder_name
|
||
|
|
return self
|
||
|
|
|
||
|
|
def ranker_name(self, ranker_name: str) -> "ModelParamBuilder":
|
||
|
|
self._model_param.ranker_name = ranker_name
|
||
|
|
return self
|
||
|
|
|
||
|
|
def filter_name(self, filter_name: str) -> "ModelParamBuilder":
|
||
|
|
self._model_param.filter_name = filter_name
|
||
|
|
return self
|
||
|
|
|
||
|
|
def booster_name(self, booster_name: str) -> "ModelParamBuilder":
|
||
|
|
self._model_param.booster_name = booster_name
|
||
|
|
return self
|
||
|
|
|
||
|
|
def passage_language(self, passage_language: str) -> "ModelParamBuilder":
|
||
|
|
self._model_param.passage_language = passage_language
|
||
|
|
return self
|
||
|
|
|
||
|
|
def build(self) -> "ModelParam":
|
||
|
|
return self._model_param
|