Skip to content
Snippets Groups Projects
Commit efa66531 authored by Alexander Heidelbach's avatar Alexander Heidelbach
Browse files

Remove ext model handli0ng from BaseModel

parent ff4e7f8f
Branches
No related tags found
No related merge requests found
......@@ -23,13 +23,11 @@ class BaseModel(ABC):
self._modelparameters = {} # type: Dict[str, zfit.Parameter]
self._covariance = {} # type: CovarianceType
self._model = None # type: Optional[zfit.core.basepdf.BasePDF]
self._ext_model = None # type: Optional[Union[Union[zfit.core.basepdf.BasePDF,zfit.core.interfaces.ZfitPDF] , Fitparameter]]
def __str__(self) -> str:
string = "Model Wrapper"
string += f"\nParameters:\n{self.parameters}"
string += f"\nModel:\n{self.model}"
string += f"\nExtended Model:\n{self.ext_model}"
return string
......@@ -61,26 +59,6 @@ class BaseModel(ABC):
def model(self) -> Optional[zfit.core.basepdf.BasePDF]:
pass
@property
def ext_model(
self,
) -> Optional[
Union[
Union[zfit.core.basepdf.BasePDF, zfit.core.interfaces.ZfitPDF], Fitparameter
]
]:
return self._ext_model
@ext_model.setter
def ext_model(self, param: Fitparameter) -> None:
assert self.model is not None
_yield = zfit.Parameter(**param.parameter_info_dict)
ext_model = self.model.create_extended(_yield)
self._ext_model = ext_model
self.modelparameters.update({str(_yield.name): _yield})
def get_fitparameter(self, name: str) -> Fitparameter:
OutputParameter = Fitparameter(name)
if self.parameters:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment