Skip to content

Commit b77d983

Browse files
committed
Fix name generation for unknown models
1 parent 39a09aa commit b77d983

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

json_to_models/registry.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,12 +206,14 @@ def fix_name_duplicates(self):
206206
"""
207207
counter = defaultdict(int)
208208
for model in self.models:
209-
counter[model.name] += 1
209+
counter[model.name or model.index] += 1
210210
if counter[model.name] > 1:
211211
model.set_raw_name(model.name_joiner(model.name, model.index), generated=True)
212212

213213
def generate_names(self):
214214
for model in self.models:
215215
if model.is_name_generated is None:
216216
model.generate_name()
217+
if model.name is None:
218+
model.set_raw_name("Unknown_" + model.index, True)
217219
self.fix_name_duplicates()

0 commit comments

Comments
 (0)