Fix debug logging (--enable-debug-logs) (#183)

Author: Obliviour

src/xpk/core/core.py

@@ -2111,8 +2111,11 @@ def get_main_container(args, system, docker_image, resource_type) -> str:
   command = args.command
   if args.enable_debug_logs:
     command = (
-        'TPU_STDERR_LOG_LEVEL=0 TPU_MIN_LOG_LEVEL=0 TF_CPP_MIN_LOG_LEVEL=0'
-        f' TPU_VMODULE=real_program_continuator=1 {args.command}'
+        'export TPU_STDERR_LOG_LEVEL=0 &&'
+        ' export TPU_MIN_LOG_LEVEL=0 &&'
+        ' export TF_CPP_MIN_LOG_LEVEL=0 &&'
+        ' export TPU_VMODULE=real_program_continuator=1 &&'
+        f' {args.command}'
     )
 
   gpu_workload_terminate_command = ''