@@ -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