@@ -46,56 +46,56 @@ impl NormalTruncated {
4646 // Threshold can probably be tuned better for performance
4747 if std_lower >= 0.5 {
4848 // One sided truncation, lower bound only
49- return Ok ( NormalTruncated ( Method :: OneSided (
49+ Ok ( NormalTruncated ( Method :: OneSided (
5050 true ,
5151 NormalTruncatedOneSided :: new ( mean, stddev, std_lower) ,
52- ) ) ) ;
52+ ) ) )
5353 } else {
5454 // We use naive rejection sampling
5555 // Also catches the case where both bounds are infinite
56- return Ok ( NormalTruncated ( Method :: Rejection (
56+ Ok ( NormalTruncated ( Method :: Rejection (
5757 NormalTruncatedRejection {
5858 normal : crate :: Normal :: new ( mean, stddev) . unwrap ( ) ,
5959 lower,
6060 upper,
6161 } ,
62- ) ) ) ;
62+ ) ) )
6363 }
6464 } else if lower == f64:: NEG_INFINITY {
6565 // Threshold can probably be tuned better for performance
6666 if std_upper <= -0.5 {
6767 // One sided truncation, upper bound only
68- return Ok ( NormalTruncated ( Method :: OneSided (
68+ Ok ( NormalTruncated ( Method :: OneSided (
6969 false ,
7070 NormalTruncatedOneSided :: new ( -mean, stddev, -std_upper) ,
71- ) ) ) ;
71+ ) ) )
7272 } else {
7373 // We use naive rejection sampling
74- return Ok ( NormalTruncated ( Method :: Rejection (
74+ Ok ( NormalTruncated ( Method :: Rejection (
7575 NormalTruncatedRejection {
7676 normal : crate :: Normal :: new ( mean, stddev) . unwrap ( ) ,
7777 lower,
7878 upper,
7979 } ,
80- ) ) ) ;
80+ ) ) )
8181 }
8282 } else {
8383 let diff = std_upper - std_lower;
8484 // Threshold can probably be tuned better for performance
8585 if diff >= 1.0 && std_lower <= 1.0 && std_upper >= -1.0 {
8686 // Naive rejection sampling
87- return Ok ( NormalTruncated ( Method :: Rejection (
87+ Ok ( NormalTruncated ( Method :: Rejection (
8888 NormalTruncatedRejection {
8989 normal : crate :: Normal :: new ( mean, stddev) . unwrap ( ) ,
9090 lower,
9191 upper,
9292 } ,
93- ) ) ) ;
93+ ) ) )
9494 } else {
9595 // Two sided truncation
96- return Ok ( NormalTruncated ( Method :: TwoSided (
96+ Ok ( NormalTruncated ( Method :: TwoSided (
9797 NormalTruncatedTwoSided :: new ( mean, stddev, std_lower, std_upper) ,
98- ) ) ) ;
98+ ) ) )
9999 }
100100 }
101101 }
0 commit comments