Skip to content

Commit c923a27

Browse files
Add params argument to export method signature
1 parent 3b3f1d7 commit c923a27

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

merlin/systems/dag/ops/hugectr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def compute_output_schema(
120120
"""
121121
return Schema([ColumnSchema("OUTPUT0", dtype=np.float32)])
122122

123-
def export(self, path, input_schema, output_schema, node_id=None, version=1):
123+
def export(self, path, input_schema, output_schema, node_id=None, params=None, version=1):
124124
"""Create and export the required config files for the hugectr model.
125125
126126
Parameters

0 commit comments

Comments
 (0)