def test_check_grads(self):
        x_ = np.array([[np.random.rand()]])

        assert check_grad(self.pi, lambda x: -self.pi(x, True)[1], x_) < 1e-5
dataset/ETHPy150Open automl/RoBO/robo/test/test_pi.py/PITestCase.test_check_grads/original