fixes a bug related to data concatenation

KanduriC · Feb 8, 2024 · 4dcf407 · 4dcf407
1 parent 326f8f1
commit 4dcf407
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/simAIRR/concatenate_repertoire_components/RepComponentConcatenation.py b/simAIRR/concatenate_repertoire_components/RepComponentConcatenation.py
@@ -58,7 +58,10 @@ def concatenate_repertoire_components(self, file_number):
             if self.components_type == "public_private":
                 concatenated_df = pd.concat(dfs_list)
             else:
-                concatenated_df = concatenate_dataframes_with_replacement(dfs_list)
+                if len(dfs_list) > 1:
+                    concatenated_df = concatenate_dataframes_with_replacement(dfs_list)
+                else:
+                    concatenated_df = pd.concat(dfs_list)
             if self.export_cdr3_aa is True:
                 concatenated_df['cdr3_aa'] = concatenated_df['junction_aa'].str[1:-1]
                 concatenated_df = concatenated_df.drop('junction_aa', axis=1)