setter

microsoft · Aug 31, 2024 · ff1aaa5 · ff1aaa5
1 parent 0ca7604
commit ff1aaa5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/onnxscript/values.py b/onnxscript/values.py
@@ -260,6 +260,7 @@ def opset(self) -> Opset: ...
     @property
     def op_schema(self) -> Optional[onnx.defs.OpSchema]: ...
 
+    @property
     def signature(self) -> Optional[_schemas.OpSignature]: ...
 
 
@@ -335,6 +336,10 @@ def signature(self) -> Optional[_schemas.OpSignature]:
         self._signature = _schemas.OpSignature.from_op_schema(self.op_schema)
         return self._signature
 
+    @signature.setter
+    def signature(self, value: _schemas.OpSignature):
+        self._signature = value
+
     @deprecation.deprecated(
         since="0.1",
         removed_in="the future",
@@ -528,6 +533,7 @@ def op_schema(self) -> Optional[onnx.defs.OpSchema]:
 
         return self._op_schema
 
+    @property
     def signature(self) -> Optional[_schemas.OpSignature]:
         """Returns the signature of this op."""
         if self._signature is not None:
@@ -541,6 +547,10 @@ def signature(self) -> Optional[_schemas.OpSignature]:
         )
         return self._signature
 
+    @signature.setter
+    def signature(self, value: _schemas.OpSignature):
+        self._signature = value
+
     def __getitem__(self, instance):
         """Returns a lambda to evaluate function using given evaluator instance.
 
@@ -672,6 +682,10 @@ def signature(self) -> Optional[_schemas.OpSignature]:
         )
         return self._signature
 
+    @signature.setter
+    def signature(self, value: _schemas.OpSignature):
+        self._signature = value
+
     @deprecation.deprecated(
         since="0.1",
         removed_in="the future",