PyPI - data-manipulation-utilities - Versions diffs - 0.2.8.dev956__tar.gz → 0.2.8.dev959__tar.gz - Mend

@@ -535,6 +535,7 @@ def pdf_to_tex(path : str, d_par : dict[str,str], skip_fixed : bool = True) -> N
 #---------------------------------------------
 def get_model(
     kind   : str,
+    nsample: int       = 1000,
     obs    : zobs|None = None,
     suffix : str|None  = None,
     lam    : float     = -0.0001) -> zpdf:
@@ -543,10 +544,11 @@ def get_model(
     Parameters:
-    kind  : 'signal' for Gaussian, 's+b' for Gaussian plus exponential
-    obs   : If provided, will use it, by default None and will be built in function
-    suffix: Optional, can be used in case multiple models are needed
-    lam   : Decay constant of exponential component, set to -0.0001 by default
+    kind   : 'signal' for Gaussian, 's+b' for Gaussian plus exponential
+    nsample: Number of entries for normalization of each component, default 1000
+    obs    : If provided, will use it, by default None and will be built in function
+    suffix : Optional, can be used in case multiple models are needed
+    lam    : Decay constant of exponential component, set to -0.0001 by default
     '''
     if suffix is not None:
         suffix = f'_{suffix}'
@@ -567,8 +569,8 @@ def get_model(
     expo= zfit.pdf.Exponential(obs=obs, lam=c)
     if kind == 's+b':
-        nexpo = zfit.param.Parameter(f'nbkg{suffix}', 1000, 0, 1000_000)
-        ngaus = zfit.param.Parameter(f'nsig{suffix}', 1000, 0, 1000_000)
+        nexpo = zfit.param.Parameter(f'nbkg{suffix}', nsample, 0, 1000_000)
+        ngaus = zfit.param.Parameter(f'nsig{suffix}', nsample, 0, 1000_000)
         bkg   = expo.create_extended(nexpo)
         sig   = gaus.create_extended(ngaus)
@@ -590,7 +592,6 @@ def get_nll(kind : str) -> ZfitLoss:
     '''
     pdf = get_model(kind=kind)
     if kind == 's+b':
         dat = pdf.create_sampler()
         return zfit.loss.ExtendedUnbinnedNLL(model=pdf, data=dat)

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: data-manipulation-utilities
-Version: 0.2.8.dev956
+Version: 0.2.8.dev959
 Summary: Project storing utilities needed to reduce boilerplate code when analyzing data
 Requires-Python: >=3.10,<3.13
 Classifier: Programming Language :: Python :: 3
@@ -1051,7 +1051,7 @@ utilities.placeholder_fit(kind='s+b', fit_dir='/some/directory')
 ### Likelihooods
 ```python
-from dmu.stats import utilities
+from dmu.stats import utilities as sut
 nll = sut.get_nll(kind='s+b')
 ```

@@ -1016,7 +1016,7 @@ utilities.placeholder_fit(kind='s+b', fit_dir='/some/directory')
 ### Likelihooods
 ```python
-from dmu.stats import utilities
+from dmu.stats import utilities as sut
 nll = sut.get_nll(kind='s+b')
 ```

@@ -1,7 +1,7 @@
 [tool.poetry]
 name           = 'data-manipulation-utilities'
 description    = 'Project storing utilities needed to reduce boilerplate code when analyzing data'
-version        = "0.2.8.dev956"
+version        = "0.2.8.dev959"
 readme         = 'README.md'
 package-mode   = true
 packages       = [

@@ -374,7 +374,7 @@ class Fitter:
         stat = res.status
-        log.info(f'{chi2:<10.3f}{pval:<10.3e}{stat:<10}')
+        log.debug(f'{chi2:<10.3f}{pval:<10.3e}{stat:<10}')
         pdf   = nll.model[0] # This class is not meant for simultaneous fits
                              # There should only be one PDF
         d_par = Fitter.get_float_pars(pdf=pdf)

data-manipulation-utilities 0.2.8.dev956__tar.gz → 0.2.8.dev959__tar.gz

data-manipulation-utilities 0.2.8.dev956tar.gz → 0.2.8.dev959tar.gz