From 31a5e572c707c3a1a4adc3e2fe350ba5109e4184 Mon Sep 17 00:00:00 2001
From: Christian Schiffer <c.schiffer@fz-juelich.de>
Date: Fri, 9 Aug 2024 11:09:42 +0200
Subject: [PATCH] Fixed threads per core

---
 atlas_server/src/app.py                               | 4 ++--
 atlas_server/src/static/training/predict.job.sh       | 1 +
 atlas_server/src/static/training/train.job.sh         | 1 +
 atlas_server/src/static/training/train_predict.job.sh | 1 +
 4 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/atlas_server/src/app.py b/atlas_server/src/app.py
index 7cdcda4..1b7da3f 100644
--- a/atlas_server/src/app.py
+++ b/atlas_server/src/app.py
@@ -84,9 +84,9 @@ class ProjectList(Resource):
     def get(self):
         # TODO This function should be hidden behind an authentication mechanism
         with Database() as db:
-            projects = db.projects.find({})
+            projects = list(db.projects.find({}))
         if not projects:
-            abort(404, f"No projects found")
+            abort(404, "No projects found")
         return _project_schema.dump(projects, many=True)
 
     @project_namespace.doc("Create a new project")
diff --git a/atlas_server/src/static/training/predict.job.sh b/atlas_server/src/static/training/predict.job.sh
index ce9663b..ff85be7 100644
--- a/atlas_server/src/static/training/predict.job.sh
+++ b/atlas_server/src/static/training/predict.job.sh
@@ -1,4 +1,5 @@
 #!/bin/bash -ex
+#SBATCH --threads-per-core=2
 
 # Load modules
 echo Loading modules...
diff --git a/atlas_server/src/static/training/train.job.sh b/atlas_server/src/static/training/train.job.sh
index 096b0ff..2a81bf1 100644
--- a/atlas_server/src/static/training/train.job.sh
+++ b/atlas_server/src/static/training/train.job.sh
@@ -1,4 +1,5 @@
 #!/bin/bash -ex
+#SBATCH --threads-per-core=2
 
 # Load modules
 echo Loading modules...
diff --git a/atlas_server/src/static/training/train_predict.job.sh b/atlas_server/src/static/training/train_predict.job.sh
index 8b9d405..40cff78 100644
--- a/atlas_server/src/static/training/train_predict.job.sh
+++ b/atlas_server/src/static/training/train_predict.job.sh
@@ -1,4 +1,5 @@
 #!/bin/bash -ex
+#SBATCH --threads-per-core=2
 
 # Load modules
 echo Loading modules...
-- 
GitLab