diff options
| -rw-r--r-- | gn3/api/correlation.py | 4 | ||||
| -rw-r--r-- | gn3/commands.py | 7 |
2 files changed, 8 insertions, 3 deletions
diff --git a/gn3/api/correlation.py b/gn3/api/correlation.py index 539b181..6bc3931 100644 --- a/gn3/api/correlation.py +++ b/gn3/api/correlation.py @@ -142,7 +142,9 @@ def partial_correlation(): conn=conn, cmd=command, job_queue=compute_job_queue(current_app), - env = {"PYTHONPATH": ":".join(sys.path), "SQL_URI": SQL_URI}, + options={ + "env": {"PYTHONPATH": ":".join(sys.path), "SQL_URI": SQL_URI}, + }, log_level=logging.getLevelName( current_app.logger.getEffectiveLevel()).lower()) return build_response({ diff --git a/gn3/commands.py b/gn3/commands.py index b4db447..73252a6 100644 --- a/gn3/commands.py +++ b/gn3/commands.py @@ -10,6 +10,7 @@ import tempfile import subprocess from datetime import datetime +from typing import Any from typing import Dict from typing import List from typing import Optional @@ -186,10 +187,12 @@ def compute_job_queue(app: Flask) -> str: def run_async_cmd( conn: Redis, job_queue: str, cmd: Union[str, Sequence[str]], - email: Optional[str] = None, log_level: str = "info", - env: Optional[dict] = None) -> str: + options: Optional[Dict[str, Any]] = None, + log_level: str = "info") -> str: """A utility function to call `gn3.commands.queue_cmd` function and run the worker in the `one-shot` mode.""" + email = options.get("email") if options else None + env = options.get("env") if options else None cmd_id = queue_cmd(conn, job_queue, cmd, email, env) worker_command = [ sys.executable, |