Merged development

phac-nml · May 31, 2022 · f0b29eb · f0b29eb
2 parents 3b6abec + 30dbe1b
commit f0b29eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/staramr/results/AMRDetectionSummary.py b/staramr/results/AMRDetectionSummary.py
@@ -37,12 +37,12 @@ def __init__(self, files, resfinder_dataframe: DataFrame, quality_module_datafra
 
     def _compile_results(self, resistance_frame: DataFrame) -> DataFrame:
         df_summary = resistance_frame.sort_values(by=['Gene']).groupby(['Isolate ID']).aggregate(
-            lambda x: {'Gene': (self.SEPARATOR + ' ').join(x.get('Gene'))})
+            {'Gene': lambda x: (self.SEPARATOR + ' ').join(x)})
         return df_summary[['Gene']].copy()
 
     def _compile_plasmids(self, plasmid_frame: DataFrame) -> DataFrame:
         ds_summary = plasmid_frame.sort_values(by=['Gene']).groupby(['Isolate ID']).aggregate(
-            lambda x: {'Gene': (self.SEPARATOR + ' ').join(x.get('Gene'))})
+            {'Gene': lambda x: (self.SEPARATOR + ' ').join(x)})
 
         ds_frame = ds_summary[['Gene']].copy()