Просмотр исходного кода

Forgot some return statements (A).

Taddeus Kroes 14 лет назад
Родитель
Сommit
8473cbe5bd
1 измененных файлов с 7 добавлено и 2 удалено
  1. 7 2
      src/create_characters.py

+ 7 - 2
src/create_characters.py

@@ -56,7 +56,10 @@ def load_learning_set(neighbours, blur_scale, verbose=0):
         if verbose:
         if verbose:
             print 'Learning set:', [c.value for c in learning_set]
             print 'Learning set:', [c.value for c in learning_set]
     else:
     else:
-        return generate_sets(neighbours, blur_scale, verbose=verbose)[0]
+        learning_set = generate_sets(neighbours, blur_scale, \
+                verbose=verbose)[0]
+
+    return learning_set
 
 
 
 
 def load_test_set(neighbours, blur_scale, verbose=0):
 def load_test_set(neighbours, blur_scale, verbose=0):
@@ -71,7 +74,9 @@ def load_test_set(neighbours, blur_scale, verbose=0):
         if verbose:
         if verbose:
             print 'Test set:', [c.value for c in test_set]
             print 'Test set:', [c.value for c in test_set]
     else:
     else:
-        return generate_sets(neighbours, blur_scale, verbose=verbose)[1]
+        test_set = generate_sets(neighbours, blur_scale, verbose=verbose)[1]
+
+    return test_set
 
 
 
 
 def generate_sets(neighbours, blur_scale, verbose=0):
 def generate_sets(neighbours, blur_scale, verbose=0):