From 4e2bde1e6ae1551e3bc3d407ad08f022c6fc7b1c Mon Sep 17 00:00:00 2001
From: Tim Repke <repke@mcc-berlin.net>
Date: Tue, 30 Jul 2024 19:53:01 +0200
Subject: [PATCH] save tracker batch size

---
 server/api/routes/evaluation.py | 3 +--
 server/util/config.py           | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/server/api/routes/evaluation.py b/server/api/routes/evaluation.py
index b4157d3..2983433 100644
--- a/server/api/routes/evaluation.py
+++ b/server/api/routes/evaluation.py
@@ -116,7 +116,6 @@ async def save_tracker(tracker: AnnotationTrackerModel,
 @router.post('/tracking/refresh', response_model=AnnotationTrackerModel)
 async def update_tracker(tracker_id: str,
                          background_tasks: BackgroundTasks,
-                         batch_size: int | None = None,
                          reset: bool = False,
                          permissions: UserPermissions = Depends(UserPermissionChecker('annotations_edit'))) \
         -> AnnotationTrackerModel:
@@ -144,7 +143,7 @@ async def update_tracker(tracker_id: str,
         await session.flush()
 
         # We are not handing over the existing tracker ORM, because the session is not persistent
-        background_tasks.add_task(bg_populate_tracker, tracker_id, batch_size, diff)
+        background_tasks.add_task(bg_populate_tracker, tracker_id, tracker.batch_size, diff)
 
         return AnnotationTrackerModel.model_validate(tracker.__dict__)
 
diff --git a/server/util/config.py b/server/util/config.py
index 2362128..3a115fd 100644
--- a/server/util/config.py
+++ b/server/util/config.py
@@ -165,7 +165,7 @@ class Settings(BaseSettings):
         if isinstance(v, dict):
             return v
         filename = info.data.get('LOG_CONF_FILE', None)
-        print(info.data)
+
         if filename is not None:
             with open(filename, 'r') as f:
                 ret = toml.loads(f.read())
-- 
GitLab