Implement getters for NPresult

LAPKB · Oct 27, 2024 · 8e30d52 · 8e30d52
1 parent 7062aef
commit 8e30d52
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "pmcore"
-version = "0.7.0"
+version = "0.7.1"
 edition = "2021"
 authors = [
     "Julián D. Otálvaro <[email protected]>",

diff --git a/src/routines/output.rs b/src/routines/output.rs
@@ -72,6 +72,18 @@ impl<E: Equation> NPResult<E> {
         self.converged
     }
 
+    pub fn get_theta(&self) -> &Array2<f64> {
+        &self.theta
+    }
+
+    pub fn get_psi(&self) -> &Array2<f64> {
+        &self.psi
+    }
+
+    pub fn get_w(&self) -> &Array1<f64> {
+        &self.w
+    }
+
     pub fn write_outputs(&self) -> Result<()> {
         if self.settings.output.write {
             let idelta: f64 = self.settings.predictions.idelta;