SET_IO PTR REMOVED FROM PROB

mathematicalmichael · Apr 30, 2019 · 40a7267 · 40a7267
1 parent 574c695
commit 40a7267
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/bet/calculateP/calculateP.py b/bet/calculateP/calculateP.py
@@ -42,8 +42,8 @@ def prob_on_emulated_samples(discretization, globalize=True):
     discretization._emulated_input_sample_set.check_num()
 
     # Check for necessary properties
-    if discretization._io_ptr_local is None:
-        discretization.set_io_ptr(globalize=True)
+#     if discretization._io_ptr_local is None:
+#         discretization.set_io_ptr(globalize=True)
     if discretization._emulated_ii_ptr_local is None:
         discretization.set_emulated_ii_ptr(globalize=False)
 
@@ -85,8 +85,8 @@ def prob(discretization, globalize=True):
     op_num = discretization._output_probability_set.check_num()
 
     # Check for necessary attributes
-    if discretization._io_ptr_local is None:
-        discretization.set_io_ptr(globalize=False)
+#     if discretization._io_ptr_local is None:
+#         discretization.set_io_ptr(globalize=False)
 
     # Calculate Probabilities
     if discretization._input_sample_set._values_local is None:

diff --git a/bet/sample.py b/bet/sample.py
@@ -2339,6 +2339,8 @@ def __init__(self, input_sample_set, output_sample_set,
 
         if output_sample_set is not None:
             self.check_nums()
+            if output_probability_set is not None:
+                self.set_io_ptr()
         else:
             logging.info("No output_sample_set")