doc: (re)add repr

maxibor · Jul 23, 2024 · 97ba7a4 · 97ba7a4
1 parent fd54950
commit 97ba7a4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pydamage/damage.py b/pydamage/damage.py
@@ -36,10 +36,9 @@ def __init__(self, reference, al_handle, wlen, g2a, subsample):
         self.wlen = wlen
         self.g2a = g2a
         self.subsample = subsample
-        # self.alignments = al_file
 
     def __repr__(self):
-        return(f"Reference {reference}")
+        return(f"Reference {self.reference}")
 
     def get_damage(self, show_al):
         """Compute CtoT substitutions
@@ -68,7 +67,7 @@ def get_damage(self, show_al):
             iterator = self.alignments
         for al in iterator:
             if al.is_unmapped is False and (
-                self.subsample is None or np.random.rand() < self.subsample
+                not self.subsample or np.random.rand() < self.subsample
             ):
                 all_damage = damage_al(
                     reference=al.get_reference_sequence(),