Extra test

temporian/core/operators/base.py

@@ -186,7 +186,9 @@ def check(self) -> None:
                         " inputs."
                     )
 
-                if available_input in [v.key for v in definition.inputs]:
+                if available_input in [
+                    v.key for v in definition.inputs if v.HasField("key")
+                ]:
                     if num_multi_input_matches != 0:
                         raise ValueError(
                             f'Input "{available_input}" matches both a prefix'

temporian/core/test/operator_test.py

@@ -102,7 +102,7 @@ def build_fake_node():
 
         t.check()
 
-    def test_check_operator_with_key_prefix_and_invalid_def(self):
+    def test_check_operator_error_overlapping_prefixes(self):
         class ToyOperator(base.Operator):
             @classmethod
             def build_op_definition(cls) -> pb.OperatorDef:
@@ -127,6 +127,33 @@ def build_fake_node():
         ):
             t.check()
 
+    def test_check_operator_error_overlapping_prefix_and_non_prefix(
+        self,
+    ):
+        class ToyOperator(base.Operator):
+            @classmethod
+            def build_op_definition(cls) -> pb.OperatorDef:
+                return pb.OperatorDef(
+                    key="TOY",
+                    inputs=[
+                        pb.OperatorDef.Input(key_prefix="input_"),
+                        pb.OperatorDef.Input(key="input_1"),
+                    ],
+                )
+
+            def _get_pandas_implementation(self):
+                raise NotImplementedError()
+
+        def build_fake_node():
+            return input_node(features=[])
+
+        t = ToyOperator()
+        t.add_input("input_1", build_fake_node())
+        with self.assertRaisesRegex(
+            ValueError, "matches both a prefix and non-prefix input"
+        ):
+            t.check()
+
 
 if __name__ == "__main__":
     absltest.main()