diff --git a/test/local_test_pet.py b/test/local_test_pet.py index f043d16a5c7a9d7d45db8ce91864a8c5325876b9..920aea654c6afc159e4ff1fab031b1c29ea69528 100644 --- a/test/local_test_pet.py +++ b/test/local_test_pet.py @@ -51,7 +51,7 @@ if __name__ == "__main__": dest="devices", help="GPU id of running. if more than one, use spacing to separate.", nargs="+", - default=0, + default=[0], type=int) args = parser.parse_args()