Skip to content

Commit 106d496

Browse files
Models with circular refs
1 parent fd805ba commit 106d496

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

pydatalab/src/pydatalab/models/__init__.py

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,13 @@
2626
"ITEM_MODELS",
2727
)
2828

29-
Sample.model_rebuild()
30-
StartingMaterial.model_rebuild()
31-
Cell.model_rebuild()
32-
Equipment.model_rebuild()
33-
Collection.model_rebuild()
29+
MODELS_WITH_CIRCULAR_REFS = [
30+
*ITEM_MODELS.values(),
31+
Collection,
32+
]
33+
34+
for model in MODELS_WITH_CIRCULAR_REFS:
35+
try:
36+
model.model_rebuild()
37+
except Exception as e:
38+
print(f"Warning: Failed to rebuild {model.__name__}: {e}")

0 commit comments

Comments
 (0)