[pyschlandals] get distribution values

aia-uclouvain · Aug 14, 2023 · 01b316c · 01b316c
1 parent f63a97e
commit 01b316c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pyschlandals/src/lib.rs b/pyschlandals/src/lib.rs
@@ -211,6 +211,14 @@ impl PyDac {
         self.nodes[node].distribution_input[index]
     }
 
+    pub fn get_distribution_number_value(&self, distribution: usize) -> usize {
+        self.distributions[distribution].probabilities.len()
+    }
+
+    pub fn get_distribution_probability(&self, distribution: usize, probability_index: usize) -> f64 {
+        self.distributions[distribution].probabilities[probability_index]
+    }
+
     pub fn to_graphviz(&self, path: String) {
 
         let mut out = String::new();