fixed the shap h5 save bug that that wrongly saved a shap_scores[0] e…

…vent when it is an array instead of an array, thereby wrongly broadcasting the same value for all the rows in the h5

bpnet/cli/shap_scores.py

@@ -330,7 +330,11 @@ def data_func(model_inputs):
 
     # save the hyp shap scores, one hot sequences & chrom positions
     # to a HDF5 file
-    save_scores(peaks_df, X, counts_shap_scores[0], output_fname)
+    if isinstance(counts_shap_scores, list):
+        save_scores(peaks_df, X, counts_shap_scores[0], output_fname)
+    else:
+        save_scores(peaks_df, X, counts_shap_scores, output_fname)
+
 
     logging.info("Generating 'profile' shap scores")
     profile_shap_scores = profile_model_profile_explainer.shap_values(
@@ -342,8 +346,12 @@ def data_func(model_inputs):
 
     # save the profile hyp shap scores, one hot sequences & chrom 
     # positions to a HDF5 file
-    save_scores(peaks_df, X, profile_shap_scores[0], output_fname)
 
+    if isinstance(counts_shap_scores, list):
+        save_scores(peaks_df, X, profile_shap_scores[0], output_fname)
+    else:
+        save_scores(peaks_df, X, profile_shap_scores, output_fname)
+
     # save the dataframe as a new .bed file 
     peaks_df.to_csv('{}/peaks_valid_scores.bed'.format(shap_dir), 
                            sep='\t', header=False, index=False)