Switch to eprint for vebosity message in InflationSDP and InflationLP

Author: eliewolfe

inflation/lp/InflationLP.py

@@ -169,13 +169,13 @@ def __init__(self,
             self._boolvec_for_FR_eqs = self.blank_bool_vec
 
         if self.verbose > 1:
-            print("Number of single operator measurements per party:", end="")
+            eprint("Number of single operator measurements per party:", end="")
             prefix = " "
             for i, measures in enumerate(inflationproblem.measurements):
                 op_count = np.prod(measures.shape[:2])
-                print(prefix + f"{self.names[i]}={op_count}", end="")
+                eprint(prefix + f"{self.names[i]}={op_count}", end="")
                 prefix = ", "
-            print()
+            eprint()
         self.use_lpi_constraints = False
 
         self.identity_operator = np.empty((0, self._nr_properties),
@@ -971,7 +971,7 @@ def write_to_file(self, filename: str) -> None:
         # Write file according to the extension
         args = self._prepare_solver_arguments(separate_bounds=True)
         if self.verbose > 0:
-            print("Writing the LP program to", filename)
+            eprint("Writing the LP program to", filename)
         if extension == "lp":
             write_to_lp(args, filename)
         elif extension == "mps":
@@ -1141,7 +1141,7 @@ def _sanitise_monomial(self, mon: Any) -> CompoundMoment:
             try:
                 return self.monomial_from_name[mon]
             except KeyError:
-                print(f"As of now we only recognize \n{list(self.monomial_from_name.keys())}")
+                eprint(f"As of now we only recognize \n{list(self.monomial_from_name.keys())}")
                 return self._sanitise_monomial(self._interpret_name(mon))
         elif isinstance(mon, Real):
             if np.isclose(float(mon), 1):
@@ -1320,7 +1320,7 @@ def _generate_lp(self) -> None:
                 orbits_non_CG, return_index=True, return_inverse=True)
             self.num_non_CG = len(old_reps_non_CG)
             if self.verbose > 1:
-                print(f"Orbits discovered! {self.num_CG} unique monomials.")
+                eprint(f"Orbits discovered! {self.num_CG} unique monomials.")
             # Obtain the real generating monomials after accounting for symmetry
         else:
             self.num_CG = self.raw_n_columns
@@ -1405,7 +1405,7 @@ def _generate_lp(self) -> None:
 
         self._lp_has_been_generated = True
         if self.verbose > 1:
-            print("LP initialization complete, ready to accept further specifics.")
+            eprint("LP initialization complete, ready to accept further specifics.")
 
     def _template_to_event_boolarray(self, template: List[int], decompressor: List[np.ndarray]) -> np.ndarray:
         if template:

inflation/sdp/InflationSDP.py

@@ -101,17 +101,17 @@ def __init__(self,
 
         self.measurements = self._generate_parties()
         if self.verbose > 1:
-            print("Number of single operator measurements per party:", end="")
+            eprint("Number of single operator measurements per party:", end="")
             prefix = " "
             for i, measures in enumerate(self.measurements):
                 counter = count()
                 deque(zip(chain.from_iterable(
                     chain.from_iterable(measures)),
                           counter),
                       maxlen=0)
-                print(prefix + f"{self.names[i]}={next(counter)}", end="")
+                eprint(prefix + f"{self.names[i]}={next(counter)}", end="")
                 prefix = ", "
-            print()
+            eprint()
         self.use_lpi_constraints = False
         self.network_scenario    = inflationproblem.is_network
         self._is_knowable_q_non_networks = \
@@ -403,7 +403,7 @@ def generate_relaxation(self,
         self.first_free_idx = first_free_index
         if self.n_vars < old_num_vars:
             if self.verbose > 0:
-                print("Further variable reduction has been made possible. Number of variables in the SDP:",
+                eprint("Further variable reduction has been made possible. Number of variables in the SDP:",
                        self.n_vars)
         # self.compmoment_from_idx = dict(zip(range(self.n_vars), monomials_as_list))
         # self.compmoment_to_idx = dict(zip(monomials_as_list, range(self.n_vars)))
@@ -1411,7 +1411,7 @@ def write_to_file(self, filename: str) -> None:
 
         # Write file according to the extension
         if self.verbose > 0:
-            print("Writing the SDP program to", filename)
+            eprint("Writing the SDP program to", filename)
         if extension == "dat-s":
             write_to_sdpa(self, filename)
         elif extension == "csv":
@@ -1864,7 +1864,7 @@ def _build_cols_from_specs(self, col_specs: List[List[int]]) -> List:
             for specs in col_specs:
                 to_print.append("1" if specs == []
                                 else "".join([self.names[p] for p in specs]))
-            print("Column structure:", "+".join(to_print))
+            eprint("Column structure:", "+".join(to_print))
 
         _zero_lexorder = np.array([0], dtype=np.intc)
         columns      = []
@@ -2082,14 +2082,14 @@ def _cleanup_after_set_values(self) -> None:
         if self.momentmatrix_has_a_one:
             num_nontrivial_known -= 1
         if self.verbose > 1 and num_nontrivial_known > 0:
-            print("Number of variables with fixed numeric value:",
+            eprint("Number of variables with fixed numeric value:",
                   len(self.known_moments))
         if len(self.semiknown_moments):
             for k in self.known_moments.keys():
                 self.semiknown_moments.pop(k, None)
         num_semiknown = len(self.semiknown_moments)
         if self.verbose > 1 and num_semiknown > 0:
-            print(f"Number of semiknown variables: {num_semiknown}")
+            eprint(f"Number of semiknown variables: {num_semiknown}")
 
     def _reset_lowerbounds(self) -> None:
         """Reset the list of lower bounds."""