def test_remove_scan_op(): with pm.Model(): pm.Normal('n', 0, 1) inference = ADVI() buff = six.StringIO() inference.run_profiling(n=10).summary(buff) assert 'theano.scan_module.scan_op.Scan' not in buff.getvalue() buff.close()
def test_remove_scan_op(): with pm.Model(): pm.Normal('n', 0, 1) inference = ADVI() buff = io.StringIO() inference.run_profiling(n=10).summary(buff) assert 'theano.scan_module.scan_op.Scan' not in buff.getvalue() buff.close()
def test_remove_scan_op(): with pm.Model(): pm.Normal("n", 0, 1) inference = ADVI() buff = io.StringIO() inference.run_profiling(n=10).summary(buff) assert "aesara.scan.op.Scan" not in buff.getvalue() buff.close()