diff --git a/hexa/pipelines/management/commands/pipelines_runner.py b/hexa/pipelines/management/commands/pipelines_runner.py index d7ea53622..72b2df0f4 100644 --- a/hexa/pipelines/management/commands/pipelines_runner.py +++ b/hexa/pipelines/management/commands/pipelines_runner.py @@ -315,7 +315,7 @@ def run_pipeline(run: PipelineRun): "HEXA_RUN_ID": str(run.id), "HEXA_PIPELINE_NAME": run.pipeline.name, "HEXA_PIPELINE_TYPE": run.pipeline.type, - "HEXA_LOG_LEVEL": run.log_level, + "HEXA_LOG_LEVEL": str(run.log_level), } if run.pipeline.type == PipelineType.NOTEBOOK: env_vars.update({"HEXA_NOTEBOOK_PATH": run.pipeline.notebook_path}) diff --git a/hexa/pipelines/tests/test_pipelines_runner.py b/hexa/pipelines/tests/test_pipelines_runner.py index 923159533..dc2440f51 100644 --- a/hexa/pipelines/tests/test_pipelines_runner.py +++ b/hexa/pipelines/tests/test_pipelines_runner.py @@ -39,7 +39,7 @@ def test_env_vars(self, _, mock_run_pipeline_docker): "HEXA_RUN_ID": "123", "HEXA_PIPELINE_NAME": "test_pipeline", "HEXA_PIPELINE_TYPE": PipelineType.NOTEBOOK, - "HEXA_LOG_LEVEL": PipelineRunLogLevel.DEBUG, + "HEXA_LOG_LEVEL": str(PipelineRunLogLevel.DEBUG), "HEXA_NOTEBOOK_PATH": "/path/to/notebook", } mock_run_pipeline_docker.assert_called_once_with(