diff --git a/tests/utils.py b/tests/utils.py index 57a38b2..51ee18e 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,10 +1,10 @@ +import inspect import os import sys import numpy as np import torch import torch.distributed as dist from typing import Optional -import inspect def init_dist(local_rank: int, num_local_ranks: int):