Commit 8473cbe5 authored by Taddeus Kroes's avatar Taddeus Kroes

Forgot some return statements (A).

parent bcd1c3d8
...@@ -56,7 +56,10 @@ def load_learning_set(neighbours, blur_scale, verbose=0): ...@@ -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): ...@@ -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):
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment