Give more leeway for correctness testing

    tg.set_loss(loss)

    tg.fit(dataset, nb_epoch=10)

    prediction = np.squeeze(tg.predict_proba_on_batch(X))

    assert_true(np.all(np.isclose(prediction, y, atol=0.5)))

    assert_true(np.all(np.isclose(prediction, y, atol=1.5)))

  def test_multi_task_regressor(self):

    n_data_points = 20

    for i in range(2):

      y_real = ys[i].X

      y_pred = prediction[:, i, :]

      assert_true(np.all(np.isclose(y_pred, y_real, atol=0.5)))

      assert_true(np.all(np.isclose(y_pred, y_real, atol=1.5)))

  def test_no_queue(self):

    n_data_points = 20