[fix] Validate interface variable names are python-valid (#2538)

* Validate interface variable names

Signed-off-by: ddl-rliu <richard.liu@dominodatalab.com>

* Remove unused import

Signed-off-by: ddl-rliu <richard.liu@dominodatalab.com>

* Fix lint error

Signed-off-by: ddl-rliu <richard.liu@dominodatalab.com>

---------

Signed-off-by: ddl-rliu <richard.liu@dominodatalab.com>
Signed-off-by: Jan Fiedler <jan@union.ai>

flytekit/core/interface.py

@@ -70,6 +70,8 @@ def __init__(
         self._inputs: Union[Dict[str, Tuple[Type, Any]], Dict[str, Type]] = {}  # type: ignore
         if inputs:
             for k, v in inputs.items():
+                if not k.isidentifier():
+                    raise ValueError(f"Input name must be valid Python identifier: {k!r}")
                 if type(v) is tuple and len(cast(Tuple, v)) > 1:
                     self._inputs[k] = v  # type: ignore
                 else:

tests/flytekit/unit/core/test_interface.py

@@ -320,6 +320,16 @@ def z(a: int, b: str) -> typing.NamedTuple("NT", x_str=str, y_int=int):
     assert typed_interface.outputs.get("y_int").description == "description for y_int"
 
 
+def test_init_interface_with_invalid_parameters():
+    from flytekit.core.interface import Interface
+
+    with pytest.raises(ValueError, match=r"Input name must be valid Python identifier:"):
+        _ = Interface({"my.input": int}, {})
+
+    with pytest.raises(ValueError, match=r"Type names and field names must be valid identifiers:"):
+        _ = Interface({}, {"my.output": int})
+
+
 def test_parameter_change_to_pickle_type():
     ctx = context_manager.FlyteContext.current_context()