fix XRayDiffractionComponent AttributeError NoneType has no attribute…

… sites in self.from_data(structure).sites
materialsproject · Jun 28, 2024 · a4140a6 · a4140a6
1 parent 9d0f896
commit a4140a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crystal_toolkit/components/diffraction.py b/crystal_toolkit/components/diffraction.py
@@ -503,8 +503,8 @@ def pattern_from_struct(struct, rad_source):
             Output(self.id("large_cell_note"), "children"),
             Input(self.id("structure"), "data"),
         )
-        def update_message(structure):
-            if len(self.from_data(structure).sites) < SITES_LIMIT:
+        def update_message(structure: Structure | None):
+            if structure is None or len(self.from_data(structure).sites) < SITES_LIMIT:
                 return html.Div([])
             return MessageContainer(
                 MessageBody(