Skip to content

Commit 538bfa4

Browse files
jeanschmidtfacebook-github-bot
authored andcommitted
Fixes test 'torchrec/modules/tests:test_lazy_extension - (#3156)
Summary: Pull Request resolved: #3156 Fixes test 'torchrec/modules/tests:test_lazy_extension - torchrec.modules.tests.test_lazy_extension.TestLazyModuleExtensionMixin: test_source_code_parity_on_infer_parameters' in accordance with D77726092 Why this would even be covered in a test? ewww (obs developed by devmate) Reviewed By: atalman Differential Revision: D77744509 fbshipit-source-id: 5403a91b3b458d073ac865a80b38b2bf380687ac
1 parent 6032436 commit 538bfa4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchrec/modules/lazy_extension.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def init_weights(m):
168168
def _infer_parameters(self: _LazyExtensionProtocol, module, args, kwargs) -> None:
169169
r"""Infers the size and initializes the parameters according to the provided input batch.
170170
171-
Given a module that contains parameters that were declared inferrable
171+
Given a module that contains parameters that were declared inferable
172172
using :class:`torch.nn.parameter.ParameterMode.Infer`, runs a forward pass
173173
in the complete module using the provided input to initialize all the parameters
174174
as needed.

0 commit comments

Comments
 (0)