Fix boolean support in cmd_conf util (#3198)

Summary:
Pull Request resolved: #3198

Fixed and extended the boolean argument parsing in the cmd_conf decorator utility.

Reviewed By: iamzainhuda

Differential Revision: D78303765

fbshipit-source-id: 441226586d2a104cc6d0fb5de15f09587ec21bd8

Author: SSYernar

torchrec/distributed/benchmark/benchmark_utils.py

@@ -524,6 +524,9 @@ def wrapper() -> Any:
                 if origin in (list, List):
                     elem_type = get_args(ftype)[0]
                     arg_kwargs.update(nargs="*", type=elem_type)
+                elif ftype is bool:
+                    # Special handling for boolean arguments
+                    arg_kwargs.update(type=lambda x: x.lower() in ["true", "1", "yes"])
                 else:
                     arg_kwargs.update(type=ftype)