Skip to content

Commit dd2db49

Browse files
authored
Fix argument parsing bug
Differential Revision: D48384586 Pull Request resolved: #756
1 parent b3c4133 commit dd2db49

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

torchx/specs/builders.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ def materialize_appdef(
135135
parameter_type = parameter.annotation
136136
parameter_type = decode_optional(parameter_type)
137137
if is_bool(parameter_type):
138-
arg_value = arg_value.lower() == "true"
138+
arg_value = arg_value and arg_value.lower() == "true"
139139
elif not is_primitive(parameter_type):
140140
arg_value = decode_from_string(arg_value, parameter_type)
141141
if parameter.kind == inspect.Parameter.VAR_POSITIONAL:

torchx/specs/test/builders_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def test_fn_with_bool(flag: bool = False) -> AppDef:
6363
return get_dummy_application("trainer-without-flag")
6464

6565

66-
def test_fn_with_bool_opional(flag: Optional[bool] = None) -> AppDef:
66+
def test_fn_with_bool_optional(flag: Optional[bool] = None) -> AppDef:
6767
"""Dummy app with or without flag
6868
6969
Args:
@@ -309,7 +309,7 @@ def test_bool_false(self) -> None:
309309

310310
def test_bool_none(self) -> None:
311311
app_def = materialize_appdef(
312-
test_fn_with_bool,
312+
test_fn_with_bool_optional,
313313
[],
314314
)
315315
self.assertEqual("trainer-without-flag", app_def.roles[0].name)

0 commit comments

Comments
 (0)