Skip to content

Commit 58bb79c

Browse files
Noah ShinNoah Shin
authored andcommitted
Converting reference_dist lambdas to defs for CI check
1 parent 1f96a43 commit 58bb79c

1 file changed

Lines changed: 23 additions & 6 deletions

File tree

tests/distributions/test_continuous.py

Lines changed: 23 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2017,7 +2017,10 @@ def halfstudentt_rng_fn(self, df, loc, scale, size, rng):
20172017
pymc_dist_params = {"nu": 5.0, "sigma": 2.0}
20182018
expected_rv_op_params = {"nu": 5.0, "sigma": 2.0}
20192019
reference_dist_params = {"df": 5.0, "loc": 0, "scale": 2.0}
2020-
reference_dist = lambda self: ft.partial(self.halfstudentt_rng_fn, rng=self.get_random_state())
2020+
2021+
def reference_dist(self):
2022+
return ft.partial(self.halfstudentt_rng_fn, rng=self.get_random_state())
2023+
20212024
checks_to_run = [
20222025
"check_pymc_params_match_rv_op",
20232026
"check_pymc_draws_match_reference",
@@ -2253,7 +2256,10 @@ def logit_normal_rng_fn(self, rng, size, loc, scale):
22532256
pymc_dist_params = {"mu": 5.0, "sigma": 10.0}
22542257
expected_rv_op_params = {"mu": 5.0, "sigma": 10.0}
22552258
reference_dist_params = {"loc": 5.0, "scale": 10.0}
2256-
reference_dist = lambda self: ft.partial(self.logit_normal_rng_fn, rng=self.get_random_state())
2259+
2260+
def reference_dist(self):
2261+
return ft.partial(self.logit_normal_rng_fn, rng=self.get_random_state())
2262+
22572263
checks_to_run = [
22582264
"check_pymc_params_match_rv_op",
22592265
"check_pymc_draws_match_reference",
@@ -2324,7 +2330,10 @@ class TestBeta(BaseTestDistributionRandom):
23242330
expected_rv_op_params = {"alpha": 2.0, "beta": 5.0}
23252331
reference_dist_params = {"a": 2.0, "b": 5.0}
23262332
size = 15
2327-
reference_dist = lambda self: ft.partial(clipped_beta_rvs, random_state=self.get_random_state())
2333+
2334+
def reference_dist(self):
2335+
return ft.partial(clipped_beta_rvs, random_state=self.get_random_state())
2336+
23282337
checks_to_run = [
23292338
"check_pymc_params_match_rv_op",
23302339
"check_pymc_draws_match_reference",
@@ -2403,7 +2412,10 @@ def halfcauchy_rng_fn(self, scale, size, rng):
24032412
pymc_dist_params = {"beta": 5.0}
24042413
expected_rv_op_params = {"beta": 5.0}
24052414
reference_dist_params = {"scale": 5.0}
2406-
reference_dist = lambda self: ft.partial(self.halfcauchy_rng_fn, rng=self.get_random_state())
2415+
2416+
def reference_dist(self):
2417+
return ft.partial(self.halfcauchy_rng_fn, rng=self.get_random_state())
2418+
24072419
checks_to_run = [
24082420
"check_pymc_params_match_rv_op",
24092421
"check_pymc_draws_match_reference_not_numba",
@@ -2545,7 +2557,10 @@ def polyagamma_rng_fn(self, size, h, z, rng):
25452557
pymc_dist_params = {"h": 1.0, "z": 0.0}
25462558
expected_rv_op_params = {"h": 1.0, "z": 0.0}
25472559
reference_dist_params = {"h": 1.0, "z": 0.0}
2548-
reference_dist = lambda self: ft.partial(self.polyagamma_rng_fn, rng=self.get_random_state())
2560+
2561+
def reference_dist(self):
2562+
return ft.partial(self.polyagamma_rng_fn, rng=self.get_random_state())
2563+
25492564
checks_to_run = [
25502565
"check_pymc_params_match_rv_op",
25512566
"check_pymc_draws_match_reference",
@@ -2566,7 +2581,9 @@ def interpolated_rng_fn(self, size, mu, sigma, rng):
25662581
pymc_dist_params = {"x_points": x_points, "pdf_points": pdf_points}
25672582
reference_dist_params = {"mu": mu, "sigma": sigma}
25682583

2569-
reference_dist = lambda self: ft.partial(self.interpolated_rng_fn, rng=self.get_random_state())
2584+
def reference_dist(self):
2585+
return ft.partial(self.interpolated_rng_fn, rng=self.get_random_state())
2586+
25702587
checks_to_run = [
25712588
"check_rv_size",
25722589
"check_draws",

0 commit comments

Comments
 (0)