lint

pinellolab · Sep 11, 2023 · 079e1d2 · 079e1d2
1 parent 6b8f5e6
commit 079e1d2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/dnadiffusion/metrics/metrics.py b/src/dnadiffusion/metrics/metrics.py
@@ -69,7 +69,7 @@ def kl_comparison_generated_sequences(
             group_number=group_compare,
             cond_weight_to_metric=1,
             save_timesteps=False,
-            save_dataframe=False
+            save_dataframe=False,
         )
         for k in use_cell_list:
             v = dict_target_cells[conditional_numeric_to_tag[k]]
@@ -78,6 +78,7 @@ def kl_comparison_generated_sequences(
         final_comp_kl.append(comp_array)
     return final_comp_kl
 
+
 def generate_heatmap(df_heat: pd.DataFrame, x_label: str, y_label: str, cell_components: str):
     plt.clf()
     plt.rcdefaults()

diff --git a/src/dnadiffusion/utils/sample_util.py b/src/dnadiffusion/utils/sample_util.py
@@ -74,4 +74,4 @@ def create_sample(
     df_results_syn["motifs"] = df_results_syn[8].apply(lambda x: x.split('motif_name "')[1].split('"')[0])
     df_results_syn[0] = df_results_syn[0].apply(lambda x: "_".join(x.split("_")[:-1]))
     df_motifs_count_syn = df_results_syn[[0, "motifs"]].groupby("motifs").count()
-    return df_motifs_count_syn
+    return df_motifs_count_syn