Misc bug fixes (#516)

This commit is contained in:
Mark Towers
2023-05-23 15:35:49 +01:00
committed by GitHub
parent e9d9515d51
commit 22a00c2a75
8 changed files with 104 additions and 58 deletions

View File

@@ -382,7 +382,6 @@ def test_space_sample_mask(space: Space, mask, n_trials: int = 100):
expected_frequency = (
np.ones(space.shape) * np.where(mask == 2, 0.5, mask) * n_trials
)
print(expected_frequency)
observed_frequency = np.sum(samples, axis=0)
assert space.shape == expected_frequency.shape == observed_frequency.shape