Fix feature generation KeyError bug

ZwickyTransientFacility · Sep 22, 2023 · 7b6eac9 · 7b6eac9
1 parent 308c284
commit 7b6eac9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tools/generate_features.py b/tools/generate_features.py
@@ -747,6 +747,9 @@ def generate_features(
 
         except ValueError:
             feature_dict.pop(_id)
+            if _id in keep_id_list:
+                keep_id_list.remove(_id)
+                tme_dict.pop(_id)
 
     basicStats = Parallel(n_jobs=Ncore)(
         delayed(lcstats.calc_basic_stats)(id, vals['tme'])