will take over class number in apply-subset

tomotools · Dec 8, 2023 · bd86c4f · bd86c4f
1 parent 2599e02
commit bd86c4f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -22,7 +22,7 @@ dependencies = [
     "tqdm"
 ]
 
-[project.optinal-dependencies]
+[project.optional-dependencies]
 dev = [
     "ruff",
     ]
@@ -68,4 +68,4 @@ extend-ignore = [
 
 [tool.ruff.format]
 quote-style = "double"
-indent-style = "tab"
+indent-style = "tab"
diff --git a/subtomotools/particle_operations.py b/subtomotools/particle_operations.py
@@ -194,6 +194,10 @@ def apply_subset(subset_star, subtomo_stars):
         subset_3d["rlnOriginXAngst"] = list(subset_2d["rlnOriginXAngst"])
         subset_3d["rlnOriginYAngst"] = list(subset_2d["rlnOriginXAngst"])
 
+        # if group is given, take over
+        if "rlnClassNumber" in subset_2d:
+            subset_3d["rlnClassNumber"] = list(subset_2d["rlnClassNumber"])
+
         starfile.write(
             {"optics": fullset_3d["optics"], "particles": subset_3d},
             Path(st_star).with_name(f"{Path(st_star).stem}_selected.star"),