fix stress in split

keysongkang · Sep 3, 2024 · 3319b67 · 3319b67
1 parent 0a8c2e9
commit 3319b67
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/utils.py b/scripts/utils.py
@@ -319,7 +319,8 @@ def split_son(num_split, E_gs, harmonic_F=False):
                     al_type = 'sigma'
                     )
                 )
-            stress_test.append(test_item['calculator']['stress'])
+            if 'stress' in test_item['calculator']:
+                stress_test.append(test_item['calculator']['stress'])
 
         # Save all information into data-test.npz
         npz_name = 'MODEL/data-test.npz'