Skip to content

Commit

Permalink
Merge b82040c into e2f7351
Browse files Browse the repository at this point in the history
  • Loading branch information
jonas-eschle committed Feb 14, 2019
2 parents e2f7351 + b82040c commit 3f487a7
Show file tree
Hide file tree
Showing 8 changed files with 17 additions and 17 deletions.
2 changes: 1 addition & 1 deletion tests/test_basePDF.py
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ def test_func():
test_values_tf = ztf.convert_to_tensor(test_values, dtype=zfit.settings.ztypes.float)

gauss_func = gauss_params1.as_func(norm_range=(-5, 5))
vals = gauss_func.value(test_values)
vals = gauss_func.func(test_values)
vals = zfit.run(vals)
assert True # better assertion?

Expand Down
10 changes: 5 additions & 5 deletions tests/test_operations.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,9 +58,9 @@ def test_param_func():
new_func_equivalent = func * param4

zfit.run(tf.global_variables_initializer())
result1 = zfit.run(new_func.value(x=rnd_test_values))
result1_equivalent = zfit.run(new_func_equivalent.value(x=rnd_test_values))
result2 = zfit.run(func.value(x=rnd_test_values) * param4)
result1 = zfit.run(new_func.func(x=rnd_test_values))
result1_equivalent = zfit.run(new_func_equivalent.func(x=rnd_test_values))
result2 = zfit.run(func.func(x=rnd_test_values) * param4)
np.testing.assert_array_equal(result1, result2)
np.testing.assert_array_equal(result1_equivalent, result2)

Expand All @@ -85,9 +85,9 @@ def func2_pure(x):
added_func = func1 + func2
prod_func = func1 * func2

added_values = added_func.value(rnd_test_values)
added_values = added_func.func(rnd_test_values)
true_added_values = func1_pure(rnd_test_values) + func2_pure(rnd_test_values)
prod_values = prod_func.value(rnd_test_values)
prod_values = prod_func.func(rnd_test_values)
true_prod_values = func1_pure(rnd_test_values) * func2_pure(rnd_test_values)

zfit.run(tf.global_variables_initializer())
Expand Down
2 changes: 1 addition & 1 deletion tests/test_pdfs.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,7 +138,7 @@ def test_func_sum():
zfit.run(tf.global_variables_initializer())
test_values = np.random.uniform(low=-3, high=4, size=10)
sum_gauss_as_func = sum_gauss.as_func(norm_range=False)
vals = sum_gauss_as_func.value(x=test_values)
vals = sum_gauss_as_func.func(x=test_values)
vals = zfit.run(vals)
# test_sum = sum([g.func(test_values) for g in gauss_dists])
np.testing.assert_allclose(vals, true_gaussian_sum(test_values), rtol=1e-2) # MC integral
Expand Down
2 changes: 1 addition & 1 deletion tests/test_simple_subclass.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ def _value(self, x):

gauss1 = SimpleGaussFunc(obs='obs1', mu=3, sigma=5)

value = gauss1.value(np.random.random(size=(10, 1)))
value = gauss1.func(np.random.random(size=(10, 1)))
zfit.run(value)

with pytest.raises(ValueError):
Expand Down
6 changes: 3 additions & 3 deletions zfit/core/basefunc.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,10 +22,10 @@ def __init__(self, obs=None, dtype: typing.Type = ztypes.float, name: str = "Bas
super().__init__(obs=obs, dtype=dtype, name=name, params=params)

def _func_to_integrate(self, x: ztyping.XType):
return self.value(x=x)
return self.func(x=x)

def _func_to_sample_from(self, x):
return self.value(x=x)
return self.func(x=x)

# TODO(Mayou36): how to deal with copy properly?
def copy(self, **override_params):
Expand All @@ -41,7 +41,7 @@ def gradients(self, x: ztyping.XType, norm_range: ztyping.LimitsType = None, par
def _value(self, x):
raise NotImplementedError

def value(self, x: ztyping.XType, name: str = "value") -> ztyping.XType: # TODO(naming): rename to `func`?
def func(self, x: ztyping.XType, name: str = "value") -> ztyping.XType: # TODO(naming): rename to `func`?
"""The function evaluated at `x`.
Args:
Expand Down
2 changes: 1 addition & 1 deletion zfit/core/interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -366,7 +366,7 @@ def sample(self, n: int, limits: ztyping.LimitsType = None, name: str = "sample"

class ZfitFunc(ZfitModel):
@abc.abstractmethod
def value(self, x: ztyping.XType, name: str = "value") -> ztyping.XType:
def func(self, x: ztyping.XType, name: str = "value") -> ztyping.XType:
raise NotImplementedError

@abc.abstractmethod
Expand Down
4 changes: 2 additions & 2 deletions zfit/core/operations.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def multiply_param_func(param: ZfitParameter, func: ZfitFunc) -> ZfitFunc:
from ..models.functions import SimpleFunc

def combined_func(x):
return param * func.value(x=x)
return param * func.func(x=x)

params = {param.name: param}
params.update(func.params)
Expand Down Expand Up @@ -217,5 +217,5 @@ def convert_func_to_pdf(func: Union[ZfitFunc, Callable], obs=None, name=None) ->
from ..models.special import SimplePDF

name = func.name if name is None else func_name
pdf = SimplePDF(func=func.value, obs=func.obs, name=name, **func.params)
pdf = SimplePDF(func=func.func, obs=func.obs, name=name, **func.params)
return pdf
6 changes: 3 additions & 3 deletions zfit/models/functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,7 @@ def __init__(self, funcs: Iterable[ZfitFunc], obs: ztyping.ObsTypeInput = None,

def _value(self, x):
# sum_funcs = tf.add_n([func.value(x) for func in self.funcs])
funcs = [func.value(x) for func in self.funcs]
funcs = [func.func(x) for func in self.funcs]
sum_funcs = tf.accumulate_n(funcs)
return sum_funcs

Expand All @@ -68,9 +68,9 @@ def __init__(self, funcs: Iterable[ZfitFunc], obs: ztyping.ObsTypeInput = None,
super().__init__(funcs=funcs, obs=obs, name=name, **kwargs)

def _value(self, x):
value = self.funcs[0].value(x)
value = self.funcs[0].func(x)
for func in self.funcs[1:]:
value *= func.value(x)
value *= func.func(x)
return value


Expand Down

0 comments on commit 3f487a7

Please sign in to comment.