Auto-sync: 2026-06-29 11:36:27

dovla_cil/eval/maniskill_policy_rollout.py

@@ -2040,7 +2040,10 @@ def _lattice_candidate_mask(
     excluded = set(exclude_types)
     rows: list[list[bool]] = []
     for case in batch:
-        allowed = [candidate_type not in excluded for candidate_type in case.candidate_types]
+        allowed = [
+            not _candidate_type_matches_exclusion(candidate_type, excluded)
+            for candidate_type in case.candidate_types
+        ]
         if not any(allowed):
             allowed = [True] * len(case.candidate_types)
         rows.append(allowed)
@@ -2057,10 +2060,21 @@ def _lattice_candidate_strict_mask(
     excluded = set(exclude_types)
     rows: list[list[bool]] = []
     for case in batch:
-        rows.append([candidate_type not in excluded for candidate_type in case.candidate_types])
+        rows.append(
+            [
+                not _candidate_type_matches_exclusion(candidate_type, excluded)
+                for candidate_type in case.candidate_types
+            ]
+        )
     return torch.tensor(rows, dtype=torch.bool, device=device)
 
 
+def _candidate_type_matches_exclusion(candidate_type: str, excluded: set[str]) -> bool:
+    if candidate_type in excluded:
+        return True
+    return any(part in excluded for part in candidate_type.split("+"))
+
+
 def _lattice_candidate_type_bonus(
     batch: list[_RolloutCase],
     *,

logs/auto_sync_hf.log

@@ -234,3 +234,4 @@ No files have been modified since last commit. Skipping to prevent empty commit.
 No files have been modified since last commit. Skipping to prevent empty commit.
 No files have been modified since last commit. Skipping to prevent empty commit.
 No files have been modified since last commit. Skipping to prevent empty commit.
+No files have been modified since last commit. Skipping to prevent empty commit.