Inherit env variables for tpu grpc worker console script

tensorflower-gardener · tensorflow-jenkins · commit e02ad1d41440 · 2024-02-29T19:35:19.000Z
PiperOrigin-RevId: 611533191
diff --git a/tensorflow/python/tools/grpc_tpu_worker_service.py b/tensorflow/python/tools/grpc_tpu_worker_service.py
@@ -74,6 +74,7 @@ def move_file_to_systemd(service_name):
 
 def enable_start_service(service_name):
   commands = [
+      "systemctl --user import-environment",
       "systemctl --user daemon-reload",
       f"systemctl --user enable {service_name}",
       f"systemctl --user start {service_name}",
