Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: akoumpa <akoumpa@users.noreply.github.com>
  • Loading branch information
akoumpa committed Oct 16, 2024
1 parent ebd7016 commit f62777e
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 9 deletions.
14 changes: 7 additions & 7 deletions nemo/collections/common/parts/run_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,25 +12,25 @@
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import json
import subprocess
import os
import shlex

from pathlib import Path
from functools import lru_cache
from omegaconf import OmegaConf, DictConfig
import subprocess
from dataclasses import dataclass
from functools import lru_cache
from pathlib import Path

import nemo_run as run
from nemo_run.core.tunnel import LocalTunnel, SSHTunnel
from nemo_run.config import NEMORUN_HOME
from nemo_run.core.execution.docker import DockerExecutor
from nemo_run.core.execution.slurm import SlurmJobDetails
from nemo_run.core.serialization.zlib_json import ZlibJSONSerializer
from nemo_run.core.tunnel import LocalTunnel, SSHTunnel
from omegaconf import DictConfig, OmegaConf

from nemo.utils import logging


@lru_cache(maxsize=2)
def get_tunnel(**ssh_tunnel):
return SSHTunnel(**ssh_tunnel)
Expand Down
2 changes: 0 additions & 2 deletions nemo/collections/llm/t5/model/t5.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,6 @@

def t5_data_step(dataloader_iter) -> Dict[str, torch.Tensor]:
from megatron.core import parallel_state
from nemo.collections.nlp.modules.common.megatron.token_level_encoder_decoder import AttnMaskType
from nemo.collections.nlp.modules.common.megatron.utils import build_attention_mask_3d

from nemo.collections.nlp.modules.common.megatron.token_level_encoder_decoder import AttnMaskType
from nemo.collections.nlp.modules.common.megatron.utils import build_attention_mask_3d
Expand Down

0 comments on commit f62777e

Please sign in to comment.