Rewrote data_list to __iter__

2020-11-19 10:17:22 -05:00
parent 857e248034
commit 6e95ff5d9d
3 changed files with 25 additions and 22 deletions
--- a/src/structure/population.py
+++ b/src/structure/population.py
@ -149,9 +149,10 @@ class Population:
        self.fitness = fitness


-    def data_list(self):
-        """Returns a list of chromosome data lists"""
-        return [chromosome.data_list() for chromosome in self.chromosome_list]
+    def __iter__(self):
+        """Returns an iterable of chromosome iterables"""
+        for chromosome in self.chromosome_list:
+            yield iter(chromosome)


    def __repr__(self):