Merge pull request #25 from kirlf/patch-2

Update method visualize()
veeresht · Feb 12, 2019 · f39bc48 · f39bc48
2 parents 68c096a + 84b59c0
commit f39bc48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/commpy/channelcoding/convcode.py b/commpy/channelcoding/convcode.py
@@ -257,7 +257,7 @@ def visualize(self, trellis_length = 2, state_order = None,
             edge_colors = ["#9E1BE0", "#06D65D"]
 
         if state_order is None:
-            state_order = range(self.number_states)
+            state_order = list(range(self.number_states))
 
         font = "sans-serif"
         fig = plt.figure()