Fixes for 2D picture sample.

CodeReclaimers · Nov 23, 2019 · ee1384d · ee1384d
1 parent 0b4e311
commit ee1384d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/examples/picture2d/clean.bat b/examples/picture2d/clean.bat
@@ -0,0 +1,2 @@
+del genome-*.bin image-*.png
+del winning-novelty-*.png novelty-*.png
diff --git a/examples/picture2d/evolve_interactive.py b/examples/picture2d/evolve_interactive.py
@@ -14,7 +14,6 @@
 
 from multiprocessing import Pool
 import neat
-from neat.six_util import itervalues
 
 from common import eval_mono_image, eval_gray_image, eval_color_image
 
@@ -43,7 +42,7 @@ def write_config(cls, f, config):
 
     def update(self, species_set, generation):
         result = []
-        for s in itervalues(species_set.species):
+        for s in species_set.species.values():
             # If any member of the species is selected (i.e., has a fitness above zero),
             # mark the species as improved.
             for m in s.members.values():

diff --git a/examples/picture2d/evolve_novelty.py b/examples/picture2d/evolve_novelty.py
@@ -1,4 +1,4 @@
-from Pillow import Image
+from PIL import Image
 import os
 import random
 from multiprocessing import Pool