Skip to content

Commit 4345efc

Browse files
author
Hossein Kavianihamedani
committed
Fix linting issues & removed async
1 parent 401c71d commit 4345efc

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/forge/controller/provisioner.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,6 @@
1414

1515
import torch
1616

17-
from forge.controller.launcher import BaseLauncher, get_launcher
18-
from forge.env import all_env_vars, FORGE_DISABLE_METRICS
19-
from forge.types import ProcessConfig, ProvisionerConfig
20-
2117
from monarch._src.actor.actor_mesh import ActorMesh
2218
from monarch._src.actor.shape import Extent
2319

@@ -26,6 +22,10 @@
2622
from monarch.tools import commands
2723
from monarch.utils import setup_env_for_distributed
2824

25+
from forge.controller.launcher import BaseLauncher, get_launcher
26+
from forge.env import all_env_vars, FORGE_DISABLE_METRICS
27+
from forge.types import ProcessConfig, ProvisionerConfig
28+
2929
logger = logging.getLogger(__name__)
3030
logger.setLevel(logging.DEBUG)
3131

@@ -134,7 +134,7 @@ async def set_environment(proc_mesh: ProcMesh, env_vars: dict[str, str]):
134134
await env_setter.set_env.call(env_vars)
135135

136136

137-
async def get_nccl_env_vars() -> dict[str, str]:
137+
def get_nccl_env_vars() -> dict[str, str]:
138138
"""Get NCCL environment variables by detecting network interfaces."""
139139
if "NCCL_SOCKET_IFNAME" in os.environ and "NCCL_IB_DISABLE" in os.environ:
140140
return {}

0 commit comments

Comments
 (0)