Fix attribute name for torch device in argparser (#5433)

Ruo-Ping Dong · web-flow · commit 59211621fc23 · 2021-06-17T15:45:50.000-07:00
diff --git a/ml-agents/mlagents/trainers/cli_utils.py b/ml-agents/mlagents/trainers/cli_utils.py
@@ -257,7 +257,7 @@ def _create_parser() -> argparse.ArgumentParser:
     torch_conf.add_argument(
         "--torch-device",
         default=None,
-        dest="device",
+        dest="torch_device",
         action=DetectDefault,
         help='Settings for the default torch.device used in training, for example, "cpu", "cuda", or "cuda:0"',
     )

Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ def _create_parser() -> argparse.ArgumentParser:`
`257`	`257`	`torch_conf.add_argument(`
`258`	`258`	`"--torch-device",`
`259`	`259`	`default=None,`
`260`		`- dest="device",`
	`260`	`+ dest="torch_device",`
`261`	`261`	`action=DetectDefault,`
`262`	`262`	`help='Settings for the default torch.device used in training, for example, "cpu", "cuda", or "cuda:0"',`
`263`	`263`	`)`