Skip to content

Commit c130bd0

Browse files
committed
fix(generator): ensure variadic options in secondary factory methods
1 parent e28dcc4 commit c130bd0

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

  • tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/generator/op

tensorflow-core/tensorflow-core-generator/src/main/java/org/tensorflow/generator/op/ClassGenerator.java

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ enum RenderMode {
105105

106106
/**
107107
* The generated options class, or null if it doesn't have one or {@link #buildOptionsClass()} has
108-
* not been ran.
108+
* not been run.
109109
*/
110110
private TypeSpec optionsClass = null;
111111

@@ -748,6 +748,10 @@ private void buildSecondaryFactory(
748748
body.add("$T.class", defaultTypes.get(attr));
749749
} else {
750750
factoryBuilder.addParameter(param);
751+
// Checking if the parameter being added is the variadic options or not
752+
if (param.name.equals("options")) {
753+
factoryBuilder.varargs();
754+
}
751755
factoryBuilder.addJavadoc("\n@param $L $L", param.name, paramTags.get(param.name));
752756
typeVars.addAll(new ResolvedType(param.type).findGenerics());
753757
body.add("$L", param.name);

0 commit comments

Comments
 (0)