Skip to content

Commit a2b600b

Browse files
committed
lints
Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>
1 parent ba275b6 commit a2b600b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/unit_tests/training/test_config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,10 @@
1414

1515
"""Unit tests for configuration validation in nemo_lm.training.config."""
1616

17-
import pytest
1817
from typing import Any, Optional, Union
1918

19+
import pytest
20+
2021
import torch
2122
from megatron.core.distributed import DistributedDataParallelConfig
2223
from megatron.core.optimizer import OptimizerConfig

0 commit comments

Comments
 (0)