Skip to content

Commit

Permalink
Fix unused import (avoid quotes in type annotations)
Browse files Browse the repository at this point in the history
Signed-off-by: Vladimir Bataev <[email protected]>
  • Loading branch information
artbataev committed Nov 9, 2022
1 parent c15fb14 commit ea433ef
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nemo/core/classes/mixins/adapter_mixin_strategies.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@


class AbstractAdapterStrategy(ABC):
def forward(self, input: torch.Tensor, adapter: torch.nn.Module, *, module: 'AdapterModuleMixin'):
def forward(self, input: torch.Tensor, adapter: torch.nn.Module, *, module: AdapterModuleMixin):
"""
Forward method that defines how the output of the adapter should be merged with the input, or if it
should be merged at all.
Expand Down Expand Up @@ -72,7 +72,7 @@ def __init__(self, stochastic_depth: float = 0.0, l2_lambda: float = 0.0):
self.stochastic_depth = stochastic_depth
self.l2_lambda = l2_lambda

def forward(self, input: torch.Tensor, adapter: torch.nn.Module, *, module: 'AdapterModuleMixin'):
def forward(self, input: torch.Tensor, adapter: torch.nn.Module, *, module: AdapterModuleMixin):
"""
A basic strategy, comprising of a residual connection over the input, after forward pass by
the underlying adapter.
Expand Down

0 comments on commit ea433ef

Please sign in to comment.