Pandas compat

Ruibin-Liu · Sep 29, 2023 · 2926903 · 2926903
1 parent 211ec92
commit 2926903
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/moldf/write_jcsv.py b/moldf/write_jcsv.py
@@ -41,4 +41,11 @@ def write_jcsv(
     with open(file_name, "w", encoding="utf-8") as out_file:
         for key, df in data.items():
             out_file.write(f"#{key}\n")
-            out_file.write(df.to_csv(index=False, line_terminator="\n"))
+            try:
+                out_file.write(
+                    df.to_csv(index=False, lineterminator="\n")
+                )  # Pandas >= 1.5
+            except TypeError:
+                out_file.write(
+                    df.to_csv(index=False, line_terminator="\n")
+                )  # Pandas < 1.5