fix error

aphp · Apr 25, 2024 · fbf5e9c · fbf5e9c
1 parent c731ca4
commit fbf5e9c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/eds_scikit/plot/omop_teva.py b/eds_scikit/plot/omop_teva.py
@@ -7,7 +7,7 @@
 from eds_scikit.io.omop_teva_default_config import default_omop_teva_config
 from eds_scikit.plot.table_viz import reduce_table, visualize_table
 
-def save_pickle(obj, path : str):
+def save_pickle(path : str, obj):
     with open(path, "wb") as outp:
         pickle.dump(obj, outp)
 
@@ -173,10 +173,10 @@ def generate_omop_teva(
         end_date=end_date,
         mapper=config["mapper"],
     )
-    save_pickle(f"{output_dir}/note_count")
     note_count = note_count[~(note_count == 0).any(axis=1)]
     chart = visualize_table(note_count, title="note table dashboard")
-    chart.save(f"{output_dir}/note_chart.html", note_count)
+    save_pickle(f"{output_dir}/note_count", note_count)
+    chart.save(f"{output_dir}/note_chart.html")
 
     logger.info("notes processing done.")