@@ -41,7 +41,7 @@ def pytest_make_parametrize_id(config, val, argname):
4141 return f"{ argname } ={ repr (val )} "
4242
4343
44- @pytest .fixture (params = [2 , 3 ], scope = "session" )
44+ @pytest .fixture (params = [2 ], scope = "session" )
4545def batch_size (request ):
4646 return request .param
4747
@@ -94,33 +94,3 @@ def random_set(batch_size, set_size, feature_size):
9494@pytest .fixture (params = [2 , 3 ], scope = "session" )
9595def set_size (request ):
9696 return request .param
97-
98-
99- @pytest .fixture (params = ["two_moons" ], scope = "session" )
100- def simulator (request ):
101- return request .getfixturevalue (request .param )
102-
103-
104- @pytest .fixture (scope = "session" )
105- def training_dataset (simulator , batch_size ):
106- from bayesflow .datasets import OfflineDataset
107-
108- num_batches = 128
109- samples = simulator .sample ((num_batches * batch_size ,))
110- return OfflineDataset (samples , batch_size = batch_size )
111-
112-
113- @pytest .fixture (scope = "session" )
114- def two_moons (batch_size ):
115- from bayesflow .simulators import TwoMoonsSimulator
116-
117- return TwoMoonsSimulator ()
118-
119-
120- @pytest .fixture (scope = "session" )
121- def validation_dataset (simulator , batch_size ):
122- from bayesflow .datasets import OfflineDataset
123-
124- num_batches = 16
125- samples = simulator .sample ((num_batches * batch_size ,))
126- return OfflineDataset (samples , batch_size = batch_size )
0 commit comments