File tree Expand file tree Collapse file tree 1 file changed +1
-27
lines changed Expand file tree Collapse file tree 1 file changed +1
-27
lines changed Original file line number Diff line number Diff line change 1212# See the License for the specific language governing permissions and
1313# limitations under the License.
1414
15- import time
16-
1715import aesara
1816import aesara .tensor as at
1917import numpy as np
@@ -144,31 +142,7 @@ def test_convergence_checks(self):
144142 ):
145143 pm .sample_smc (draws = 99 )
146144
147- def test_parallel_sampling (self ):
148- # Cache graph
149- with self .slow_model :
150- _ = pm .sample_smc (draws = 10 , chains = 1 , cores = 1 , return_inferencedata = False )
151-
152- chains = 4
153- draws = 100
154-
155- t0 = time .time ()
156- with self .slow_model :
157- idata = pm .sample_smc (draws = draws , chains = chains , cores = 4 )
158- t_mp = time .time () - t0
159- assert idata .posterior .dims ["chain" ] == chains
160- assert idata .posterior .dims ["draw" ] == draws
161-
162- t0 = time .time ()
163- with self .slow_model :
164- idata = pm .sample_smc (draws = draws , chains = chains , cores = 1 )
165- t_seq = time .time () - t0
166- assert idata .posterior .dims ["chain" ] == chains
167- assert idata .posterior .dims ["draw" ] == draws
168-
169- assert t_mp < t_seq
170-
171- def test_depracated_parallel_arg (self ):
145+ def test_deprecated_parallel_arg (self ):
172146 with self .fast_model :
173147 with pytest .warns (
174148 DeprecationWarning ,
You can’t perform that action at this time.
0 commit comments