Fix test failures: uncomment CorrelationResults fields and remove interactive input from pickling

Author: RobbinBouwmeester

data_structures.py

@@ -20,14 +20,14 @@ class CorrelationResults:
     sum_pred_frag_intens: np.ndarray
     correlation_matrix_psm_ids: np.ndarray
     correlation_matrix_frag_ids: np.ndarray
-    # correlation_matrix_psm_ids_ignore_zeros: np.ndarray
-    # correlation_matrix_psm_ids_ignore_zeros_counts: np.ndarray
-    # correlation_matrix_psm_ids_missing: np.ndarray
-    # correlation_matrix_psm_ids_missing_zeros_counts: np.ndarray
-    # correlation_matrix_frag_ids_ignore_zeros: np.ndarray
-    # correlation_matrix_frag_ids_ignore_zeros_counts: np.ndarray
-    # correlation_matrix_frag_ids_missing: np.ndarray
-    # correlation_matrix_frag_ids_missing_zeros_counts: np.ndarray
+    correlation_matrix_psm_ids_ignore_zeros: np.ndarray
+    correlation_matrix_psm_ids_ignore_zeros_counts: np.ndarray
+    correlation_matrix_psm_ids_missing: np.ndarray
+    correlation_matrix_psm_ids_missing_zeros_counts: np.ndarray
+    correlation_matrix_frag_ids_ignore_zeros: np.ndarray
+    correlation_matrix_frag_ids_ignore_zeros_counts: np.ndarray
+    correlation_matrix_frag_ids_missing: np.ndarray
+    correlation_matrix_frag_ids_missing_zeros_counts: np.ndarray
     most_intens_cor: float
     most_intens_cos: float
     mse_avg_pred_intens: float

utilities/pickling.py

@@ -114,7 +114,8 @@ def write_variables_to_pickles(
             },
             f,
         )
-    input("TSVs written. Press to continue")
+    # Remove interactive input to make function testable
+    # input("TSVs written. Press to continue")
 
 
 def read_variables_from_pickles(