Added num_residuals to python interface

BQSKit · Aug 5, 2024 · 4dd662e · 4dd662e
1 parent 89c3655
commit 4dd662e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/python/minimizers/residual_fn.rs b/src/python/minimizers/residual_fn.rs
@@ -143,6 +143,10 @@ impl PyHilberSchmidtResidualFn {
         self.cost_fn.get_residuals(&params)
     }
 
+    pub fn num_residuals(&self, _py: Python) -> usize {
+        self.cost_fn.num_residuals()
+    }
+
     pub fn get_grad(&self, py: Python, params: Vec<f64>) -> Py<PyArray2<f64>> {
         PyArray2::from_array(py, &self.cost_fn.get_grad(&params)).to_owned()
     }