Fix linter warnings

Signed-off-by: Tim Moon <tmoon@nvidia.com>
NVIDIA · Sep 27, 2024 · df8e7b0 · df8e7b0
1 parent c973cff
commit df8e7b0
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/transformer_engine/pytorch/module/layernorm.py b/transformer_engine/pytorch/module/layernorm.py
@@ -12,7 +12,6 @@
 from torch.nn import init
 
 import transformer_engine_torch as tex
-from .base import TransformerEngineBaseModule
 from ..cpp_extensions import (
     layernorm_fwd_inf,
 )

diff --git a/transformer_engine/pytorch/module/rmsnorm.py b/transformer_engine/pytorch/module/rmsnorm.py
@@ -11,7 +11,6 @@
 from torch.nn.parameter import Parameter
 from torch.nn import init
 
-from .base import TransformerEngineBaseModule
 from .. import cpp_extensions as tex
 from ..jit import no_torch_dynamo
 from ..utils import cast_if_needed