Pass the session object to test and hint functions

1 files changed, 2 insertions, 2 deletions

diff --git a/prolog/problems/family_relations/sister_2/common.py b/prolog/problems/family_relations/sister_2/common.py
index 9be9e6d..96ff47d 100644
--- a/prolog/problems/family_relations/sister_2/common.py
+++ b/prolog/problems/family_relations/sister_2/common.py
@@ -17,7 +17,7 @@ sister(X, Y) :-
 import prolog.engine
 import server.problems
 
-def test(program):
+def test(session, program):
     # Test queries and expected answers (values of X).
     queries = [
         ('sister(melanie, X)', set(['andrew'])),
@@ -46,7 +46,7 @@ def test(program):
 
     return n_correct, len(queries)
 
-def hint(program):
+def hint(session, program):
     if '\=' not in program and '\==' not in program:
         return [{'id': 'x_y_must_be_different'}]
     return None