Skip to content

Commit 6d9b7ef

Browse files
committed
fix type in graph name generator
1 parent d769f3f commit 6d9b7ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ebm2onnx/graph.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ class Graph(NamedTuple):
1212
initializers: List[onnx.TensorProto] = []
1313

1414

15-
def create_name_generator() -> Callable[[], str]:
15+
def create_name_generator() -> Callable[[str], str]:
1616
state = {}
1717
def _generate_unique_name(name: str) -> str:
1818
""" Generates a new globaly unique name in the graph

0 commit comments

Comments
 (0)