fix: add _npt.gro to results download

LABIOQUIM · Jun 7, 2023 · 8be8f77 · 8be8f77
1 parent 66f6ec8
commit 8be8f77
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/apps/server/server/resources/downloads/results.py b/apps/server/server/resources/downloads/results.py
@@ -21,7 +21,11 @@ def get(self):
         with zipfile.ZipFile(file_results_zip, "w") as z:
             for folder, _, files in os.walk(folder_run_path):
                 for file in files:
-                    if file.endswith("_PBC.xtc") or file.endswith("_pr.tpr"):
+                    if (
+                        file.endswith("_PBC.xtc")
+                        or file.endswith("_pr.tpr")
+                        or file.endswith("_npt.gro")
+                    ):
                         z.write(
                             os.path.join(folder, file),
                             file,