fix error in transition matirx function

eshanking · May 12, 2024 · befac46 · befac46
1 parent f876bf8
commit befac46
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/fears/utils/plotter.py b/fears/utils/plotter.py
@@ -959,10 +959,9 @@ def gen_transition_matrix(p, conc=10**0, fit_land=None):
             j for j in range(N) if genotypes[j][1] > fitness_i and bin(i ^ j).count('1') == 1
         ]
 
-        if i == 0 or i == 15:
-            if len(more_fit_neighbors) == 0:
-                transition_matrix[i, i] = 1
-                continue  # Skip further processing for this genotype
+        if len(more_fit_neighbors) == 0:
+            transition_matrix[i, i] = 1
+            continue  # Skip further processing for this genotype
 
         if more_fit_neighbors:
             prob_per_neighbor = 1 / len(more_fit_neighbors)