Commit d23601d1 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

All tests passing

parent 95a60f90
......@@ -442,7 +442,7 @@ static auto m_step = bob::extension::FunctionDoc(
"",
true
)
.add_prototype("isv_base,stats")
.add_prototype("isv_base")
.add_parameter("isv_base", ":py:class:`bob.learn.misc.ISVBase`", "ISVBase Object");
static PyObject* PyBobLearnMiscISVTrainer_m_step(PyBobLearnMiscISVTrainerObject* self, PyObject* args, PyObject* kwargs) {
BOB_TRY
......@@ -452,7 +452,7 @@ static PyObject* PyBobLearnMiscISVTrainer_m_step(PyBobLearnMiscISVTrainerObject*
PyBobLearnMiscISVBaseObject* isv_base = 0;
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O!", kwlist, &PyBobLearnMiscISVBase_Type, &isv_base)) Py_RETURN_NONE;
if (!PyArg_ParseTupleAndKeywords(args, kwargs, "O!", kwlist, &PyBobLearnMiscISVBase_Type, &isv_base)) return 0;
self->cxx->mStep(*isv_base->cxx);
......
......@@ -231,7 +231,7 @@ def test_ISVTrainAndEnrol():
mb.u = M_u
for i in range(10):
t.e_step(mb, TRAINING_STATS)
t.m_step(mb, TRAINING_STATS)
t.m_step(mb)
assert numpy.allclose(mb.d, d_ref, eps)
assert numpy.allclose(mb.u, u_ref, eps)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment