fix assignment type bug

ablab · Apr 25, 2024 · 72b96ba · 72b96ba
1 parent 070c5b4
commit 72b96ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/long_read_counter.py b/src/long_read_counter.py
@@ -236,7 +236,7 @@ def add_read_info(self, read_assignment=None):
             if len(feature_ids):
                 self.ambiguous_reads += 1
 
-        elif read_assignment.assignment_type.is_inconsistent():
+        elif assignment_type.is_inconsistent():
             count_value = self.read_counter.process_inconsistent(assignment_type, len(feature_ids))
             if count_value > 0:
                 for feature_id in feature_ids:
@@ -245,7 +245,7 @@ def add_read_info(self, read_assignment=None):
                     self.all_features.add(feature_id)
                     #self.confirmed_features.add((group_id, feature_id))
 
-        elif read_assignment.assignment_type.is_unique():
+        elif assignment_type.is_unique():
             feature_id = list(feature_ids)[0]
             self.feature_counter[group_id][feature_id] += 1.0
             self.all_features.add(feature_id)