Fix pandas group by

The-Academic-Observatory · Aug 2, 2023 · b6f972b · b6f972b
1 parent da4a573
commit b6f972b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/academic_observatory_workflows/workflows/oa_web_workflow.py b/academic_observatory_workflows/workflows/oa_web_workflow.py
@@ -1472,7 +1472,7 @@ def make_entities(entity_type: str, df_index: pd.DataFrame, df_data: pd.DataFram
     total = len(df_index)
 
     logging.info(f"Making entities: {entity_type}")
-    ts_groups = df_data.groupby([key_id])
+    ts_groups = df_data.groupby(key_id)
 
     for entity_id, df_group in ts_groups:
         # Exclude countries and institutions with small num outputs