Add default values for output_len to avoid errors from lacking requir…

…ed arguments
NVIDIA · Aug 14, 2024 · b8cfee8 · b8cfee8
1 parent 29da24e
commit b8cfee8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tripy/tripy/frontend/trace/ops/expand.py b/tripy/tripy/frontend/trace/ops/expand.py
@@ -29,7 +29,7 @@
 @dataclass(repr=False)
 class Expand(BaseTraceOp):
     output_rank: int
-    output_len: Optional[int]  # only used to help with infer_len for a shape input
+    output_len: Optional[int] = None  # only used to help with infer_len for a shape input
 
     def infer_dtypes(self):
         self.outputs[0].dtype = self.inputs[0].dtype

diff --git a/tripy/tripy/frontend/trace/ops/reshape.py b/tripy/tripy/frontend/trace/ops/reshape.py
@@ -29,7 +29,7 @@
 class Reshape(BaseTraceOp):
 
     output_rank: int
-    output_len: Optional[int]  # only used to help with infer_len for a shape input
+    output_len: Optional[int] = None  # only used to help with infer_len for a shape input
 
     def infer_dtypes(self):
         self.outputs[0].dtype = self.inputs[0].dtype