From 15dd19291dd0ba917478f3ac20237781143e2f10 Mon Sep 17 00:00:00 2001 From: CodemodService Bot Date: Fri, 3 Nov 2023 03:12:50 -0700 Subject: [PATCH] ai_codesign Reviewed By: zsol Differential Revision: D50964791 --- torchrec_dlrm/dlrm_main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/torchrec_dlrm/dlrm_main.py b/torchrec_dlrm/dlrm_main.py index 99f95d01..5b02bbbd 100644 --- a/torchrec_dlrm/dlrm_main.py +++ b/torchrec_dlrm/dlrm_main.py @@ -719,5 +719,9 @@ def optimizer_with_params(): multihot.save_freqs_stats() -if __name__ == "__main__": +def invoke_main() -> None: main(sys.argv[1:]) + + +if __name__ == "__main__": + invoke_main() # pragma: no cover