change eps to mean(eff)

appletree/component.py

@@ -53,13 +53,6 @@ def __init__(self, name: Optional[str] = None, llh_name: Optional[str] = None, *
         if "bins" in kwargs.keys() and "bins_type" in kwargs.keys():
             self.set_binning(**kwargs)
 
-            if self.bins_type != "meshgrid" and self.add_eps_to_hist:
-                warn(
-                    "It is empirically dangerous to have add_eps_to_hist==True, "
-                    "when your bins_type is not meshgrid! It may lead to very bad fit with "
-                    "lots of eff==0."
-                )
-
     def set_binning(self, **kwargs):
         """Set binning of component."""
         if "bins" not in kwargs.keys() or "bins_type" not in kwargs.keys():
@@ -158,7 +151,8 @@ def implement_binning(self, mc, eff):
             raise ValueError(f"Unsupported bins_type {self.bins_type}!")
         if self.add_eps_to_hist:
             # as an uncertainty to prevent blowing up
-            hist = jnp.clip(hist, 1.0, jnp.inf)
+            # uncertainty = 1e-10 + jnp.mean(eff)
+            hist = jnp.clip(hist, 1e-10 + jnp.mean(eff), jnp.inf)
         return hist
 
     def get_normalization(self, hist, parameters, batch_size=None):