Sum memory usage stat for patient_df

opensafely-core · Apr 14, 2022 · e7c8f14 · e7c8f14
1 parent 501c2c2
commit e7c8f14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cohortextractor/cohortextractor.py b/cohortextractor/cohortextractor.py
@@ -364,7 +364,7 @@ def _generate_measures(
                             dataframe="patient_df",
                             measure_id=measure.id,
                             date=date,
-                            memory=patient_df.memory_usage(),
+                            memory=patient_df.memory_usage(deep=True).sum(),
                         )
                 with log_execution_time(
                     logger,