浏览代码

Forgot some return statements (A).

Taddeus Kroes 14 年之前
父节点
当前提交
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:
             print 'Learning set:', [c.value for c in learning_set]
     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):
@@ -71,7 +74,9 @@ def load_test_set(neighbours, blur_scale, verbose=0):
         if verbose:
             print 'Test set:', [c.value for c in test_set]
     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):