fix: force quoting when dumping runtimeEnvYAML (#297)

Author: JosefNagelschmidt

src/dagster_ray/kuberay/pipes.py

@@ -180,7 +180,7 @@ def _enrich_ray_job(
         runtime_env["env_vars"] = runtime_env.get("env_vars", {})
         runtime_env["env_vars"].update(env_vars)
 
-        ray_job["spec"]["runtimeEnvYAML"] = yaml.safe_dump(runtime_env)
+        ray_job["spec"]["runtimeEnvYAML"] = yaml.safe_dump(runtime_env, default_style='"')
 
         image_from_run_tag = context.run.tags.get("dagster/image")
 

tests/kuberay/test_pipes.py

@@ -3,6 +3,7 @@
 import dagster as dg
 import pytest
 import ray  # noqa: TID253
+import yaml
 from dagster._core.definitions.data_version import (
     DATA_VERSION_IS_USER_PROVIDED_TAG,
     DATA_VERSION_TAG,
@@ -26,6 +27,7 @@
     "spec": {
         "activeDeadlineSeconds": 10800,
         "entrypoint": ENTRYPOINT,
+        "runtimeEnvYAML": yaml.dump({"env_vars": {"FOO": "1E-5"}}, default_style='"'),
         "entrypointNumCpus": 0.1,
         "rayClusterSpec": {
             "autoscalerOptions": {