updates error message

ploomber · Feb 15, 2021 · c623e68 · c623e68
1 parent a1a74d2
commit c623e68
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/ploomber/util/util.py b/src/ploomber/util/util.py
@@ -180,7 +180,7 @@ def signature_check(fn, params, task_name):
 
     if 'upstream' in missing:
         errors.append('Verify this task declared upstream depedencies or '
-                      'remove the argument')
+                      'remove the "upstream" argument from the function')
 
     missing_except_upstream = sorted(missing - {'upstream'})
 

diff --git a/tests/test_util_signature_check.py b/tests/test_util_signature_check.py
@@ -61,7 +61,7 @@ def fn(upstream):
     error = ('Error rendering task "task" initialized with function '
              '"fn". Missing arguments: [\'upstream\']. '
              'Verify this task declared upstream depedencies or remove the '
-             'argument')
+             '"upstream" argument from the function')
     assert error == str(excinfo.value)
 
 
@@ -74,7 +74,8 @@ def fn(upstream, a):
 
     error = ('Error rendering task "task" initialized with function "fn". '
              'Missing arguments: [\'a\', \'upstream\']. Verify this '
-             'task declared upstream depedencies or remove the argument. '
+             'task declared upstream depedencies or remove the "upstream" '
+             'argument from the function. '
              'Pass [\'a\'] in "params"')
     assert error == str(excinfo.value)