single prediction bug fix

tiger.py

@@ -178,7 +178,7 @@ def get_on_target_predictions(transcripts: pd.DataFrame, model: tf.keras.Model,
         target_seq, guide_seq, model_inputs = process_data(row[SEQ_COL])
 
         # get predictions
-        lfc_estimate = np.squeeze(model.predict(model_inputs, batch_size=BATCH_SIZE_COMPUTE, verbose=False))
+        lfc_estimate = model.predict(model_inputs, batch_size=BATCH_SIZE_COMPUTE, verbose=False)[:, 0]
         lfc_estimate = calibrate_predictions(lfc_estimate, num_mismatches=np.zeros_like(lfc_estimate))
         scores = transform_predictions(lfc_estimate)
         predictions = pd.concat([predictions, pd.DataFrame({
@@ -308,7 +308,7 @@ def predict_off_target(off_targets: pd.DataFrame, model: tf.keras.Model):
         tf.reshape(one_hot_encode_sequence(off_targets[TARGET_COL], add_context_padding=False), [len(off_targets), -1]),
         tf.reshape(one_hot_encode_sequence(off_targets[GUIDE_COL], add_context_padding=True), [len(off_targets), -1]),
         ], axis=-1)
-    lfc_estimate = np.squeeze(model.predict(model_inputs, batch_size=BATCH_SIZE_COMPUTE, verbose=False))
+    lfc_estimate = model.predict(model_inputs, batch_size=BATCH_SIZE_COMPUTE, verbose=False)[:, 0]
     lfc_estimate = calibrate_predictions(lfc_estimate, off_targets['Number of Mismatches'].to_numpy())
     off_targets[SCORE_COL] = transform_predictions(lfc_estimate)