Skip to content

Commit d8ea7b8

Browse files
authored
Merge pull request #41 from phobson/fix-greek-lambda
Fix greek lambda
2 parents 017b973 + e329334 commit d8ea7b8

File tree

3 files changed

+12
-3
lines changed

3 files changed

+12
-3
lines changed

paramnormal/dist.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -887,7 +887,16 @@ class rice(BaseDist_Mixin):
887887
@utils.greco_deco
888888
def _process_args(R=None, loc=0, sigma=None, fit=False):
889889
loc_key, scale_key = utils._get_loc_scale_keys(fit=fit)
890-
return {loc_key: loc, scale_key: sigma, 'b': R/sigma if R is not None else None}
890+
891+
if fit:
892+
bkey = 'fb'
893+
else:
894+
bkey = 'b'
895+
896+
b = None
897+
if R is not None and sigma is not None:
898+
b = R / sigma
899+
return {loc_key: loc, scale_key: sigma, bkey: b}
891900

892901
@classmethod
893902
def fit(cls, data, **guesses):

paramnormal/tests/test_dist.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,7 @@ def test_processargs(self):
365365
assert result == expected
366366

367367
result = self.dist._process_args(R=10, sigma=2, fit=True)
368-
expected = dict(b=5, floc=0, fscale=2)
368+
expected = dict(fb=5, floc=0, fscale=2)
369369
assert result == expected
370370

371371
@seed

paramnormal/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
'α': 'alpha',
1111
'β': 'beta',
1212
'γ': 'gamma',
13-
'λ': 'lambda_',
13+
'λ': 'lamda',
1414
'θ': 'theta',
1515
}
1616

0 commit comments

Comments
 (0)