Skip to content

Commit ad5545d

Browse files
committed
debug macos test reproducibility issue
1 parent 5906825 commit ad5545d

File tree

4 files changed

+15
-2
lines changed

4 files changed

+15
-2
lines changed

.github/workflows/test.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ jobs:
1414
strategy:
1515
matrix:
1616
os: [ubuntu-latest, macos-latest, windows-latest]
17-
python-version: [3.9, "3.10", "3.11", "3.12"]
17+
python-version: ["3.12"]
1818

1919
timeout-minutes: 30
2020

black_it/calibrator.py

+4
Original file line numberDiff line numberDiff line change
@@ -379,12 +379,16 @@ def calibrate(self, n_batches: int) -> tuple[NDArray, NDArray]:
379379
self.params_samp,
380380
self.losses_samp,
381381
)
382+
print(f"Parameters sampled by {method.__class__}")
383+
print(new_params)
382384

383385
t_eval = time.time()
384386

385387
# simulate an ensemble of models for different parameters
386388

387389
new_simulated_data = self.simulate_model(new_params)
390+
print(f"New simulated data: ")
391+
print(new_simulated_data)
388392

389393
new_losses = []
390394

black_it/utils/seedable.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ def random_generator(self) -> np.random.Generator:
7575

7676
def _get_random_seed(self) -> int:
7777
"""Get new random seed from the current random generator."""
78-
return get_random_seed(self.__random_generator)
78+
seed = get_random_seed(self.__random_generator)
79+
print("Generated seed: ", seed)
80+
return seed
7981

8082

8183
def get_random_seed(random_generator: np.random.Generator) -> int:

examples/main.py

+7
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,10 @@
1616
# along with this program. If not, see <http://www.gnu.org/licenses/>.
1717

1818
"""This is a simple example showing the main features of the library."""
19+
import sys
20+
21+
import numpy as np
22+
1923
import models.simple_models as md
2024

2125
from black_it.calibrator import Calibrator
@@ -24,6 +28,8 @@
2428
from black_it.samplers.halton import HaltonSampler
2529
from black_it.samplers.random_forest import RandomForestSampler
2630

31+
np.set_printoptions(threshold=sys.maxsize)
32+
2733
if __name__ == "__main__":
2834
true_params = [0.20, 0.20, 0.75]
2935
bounds = [
@@ -44,6 +50,7 @@
4450
N = 2000
4551
seed = 1
4652
real_data = model(true_params, N, seed)
53+
print("Real data:", real_data)
4754

4855
# define a loss
4956
loss = MethodOfMomentsLoss()

0 commit comments

Comments
 (0)