make schedule end on nocodb end time, update timeout (#19)

simonsobs · Feb 11, 2025 · ee8b4b7 · ee8b4b7
1 parent 2809886
commit ee8b4b7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -29,4 +29,4 @@ RUN pip install .
 
 # Run server
 EXPOSE 8010
-ENTRYPOINT ["dumb-init", "gunicorn", "--bind", "0.0.0.0:8010", "--timeout", "120", "scheduler_server.app:app"]
+ENTRYPOINT ["dumb-init", "gunicorn", "--bind", "0.0.0.0:8010", "--timeout", "180", "scheduler_server.app:app"]
diff --git a/src/scheduler_server/handler.py b/src/scheduler_server/handler.py
@@ -107,7 +107,7 @@ def rest_handler(t0, t1, policy_config={}):
     module = importlib.util.module_from_spec(spec)
     spec.loader.exec_module(module)
 
-    schedule = module.main(t0=t0, t1=t1, **config)
+    schedule = module.main(t0=t0, t1=min(t1, datetime.fromisoformat(best_plan['to'])), **config)
 
     return schedule