Fix None when handling electronic temperature

Signed-off-by: zjgemi <[email protected]>
deepmodeling · Oct 13, 2024 · 5f263d3 · 5f263d3
1 parent 4d9273f
commit 5f263d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dpgen2/op/collect_data.py b/dpgen2/op/collect_data.py
@@ -95,9 +95,9 @@ def execute(
 
         ms = dpdata.MultiSystems(type_map=type_map)
         for ii in labeled_data:
-            if len(list(ii.rglob("fparam.npy"))) > 0:
+            if ii and len(list(ii.rglob("fparam.npy"))) > 0:
                 setup_ele_temp(False)
-            if len(list(ii.rglob("aparam.npy"))) > 0:
+            if ii and len(list(ii.rglob("aparam.npy"))) > 0:
                 setup_ele_temp(True)
             ss = dpdata.LabeledSystem(ii, fmt="deepmd/npy")
             ms.append(ss)