Commit 3993cbb3 by Usman Khalid

Updated docs and test data for rubric_dict.

TNL-708
parent 26f1237d
...@@ -127,7 +127,7 @@ def validate_training_examples(rubric, examples): ...@@ -127,7 +127,7 @@ def validate_training_examples(rubric, examples):
>>> ] >>> ]
>>> >>>
>>> rubric = { >>> rubric = {
>>> "prompt": "Write an essay!", >>> "prompts": [{"description": "Write an essay!"}],
>>> "criteria": [ >>> "criteria": [
>>> { >>> {
>>> "order_num": 0, >>> "order_num": 0,
...@@ -324,7 +324,7 @@ def get_training_example(submission_uuid, rubric, examples): ...@@ -324,7 +324,7 @@ def get_training_example(submission_uuid, rubric, examples):
{ {
'answer': u'Lorem ipsum', 'answer': u'Lorem ipsum',
'rubric': { 'rubric': {
"prompt": "Write an essay!", "prompts": [{"description": "Write an essay!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
...@@ -245,7 +245,7 @@ def rubric_from_dict(rubric_dict): ...@@ -245,7 +245,7 @@ def rubric_from_dict(rubric_dict):
Sample data (one criterion, two options):: Sample data (one criterion, two options)::
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
...@@ -107,7 +107,7 @@ def deserialize_training_examples(examples, rubric_dict): ...@@ -107,7 +107,7 @@ def deserialize_training_examples(examples, rubric_dict):
>>> ] >>> ]
>>> >>>
>>> rubric = { >>> rubric = {
>>> "prompt": "Write an essay!", >>> "prompts": [{"description": "Write an essay!"}],
>>> "criteria": [ >>> "criteria": [
>>> { >>> {
>>> "order_num": 0, >>> "order_num": 0,
......
...@@ -34,7 +34,7 @@ RUBRIC_OPTIONS = [ ...@@ -34,7 +34,7 @@ RUBRIC_OPTIONS = [
] ]
RUBRIC = { RUBRIC = {
'prompt': u"МоъЎ-ↁіск; оѓ, ГЂэ ЩЂаlэ", 'prompts': [{"description": u"МоъЎ-ↁіск; оѓ, ГЂэ ЩЂаlэ"}],
'criteria': [ 'criteria': [
{ {
"order_num": 0, "order_num": 0,
......
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
] ]
} }
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
{ {
"prompt": "Create a plan to deliver ora2!" "prompts": [{"description": "Create a plan to deliver ora2!"}]
} }
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
{ {
"prompt": "Create a plan to deliver ora2!", "prompts": [{"description": "Create a plan to deliver ora2!"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
......
{ {
"valid": { "valid": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -78,7 +78,7 @@ ...@@ -78,7 +78,7 @@
"missing_options": { "missing_options": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -148,7 +148,7 @@ ...@@ -148,7 +148,7 @@
"extra_options": { "extra_options": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -225,7 +225,7 @@ ...@@ -225,7 +225,7 @@
"missing_and_extra_options": { "missing_and_extra_options": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -291,7 +291,7 @@ ...@@ -291,7 +291,7 @@
"invalid_option_name": { "invalid_option_name": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -374,7 +374,7 @@ ...@@ -374,7 +374,7 @@
"rubric_missing_options_list": { "rubric_missing_options_list": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -397,7 +397,7 @@ ...@@ -397,7 +397,7 @@
"rubric_missing_criteria_list": { "rubric_missing_criteria_list": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽" "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}]
}, },
"examples": [ "examples": [
{ {
...@@ -413,7 +413,7 @@ ...@@ -413,7 +413,7 @@
"example_missing_keys": { "example_missing_keys": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -478,7 +478,7 @@ ...@@ -478,7 +478,7 @@
"feedback_only_criterion": { "feedback_only_criterion": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -518,7 +518,7 @@ ...@@ -518,7 +518,7 @@
"feedback_only_criterion_extra_score": { "feedback_only_criterion_extra_score": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 0, "order_num": 0,
...@@ -561,7 +561,7 @@ ...@@ -561,7 +561,7 @@
"feedback_only_all_criteria": { "feedback_only_all_criteria": {
"rubric": { "rubric": {
"prompt": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽", "prompts": [{"description": "𝓣𝓮𝓼𝓽 𝓹𝓻𝓸𝓶𝓹𝓽"}],
"criteria": [ "criteria": [
{ {
"order_num": 1, "order_num": 1,
......
...@@ -122,7 +122,7 @@ class AIGradingWorkflowTest(CacheResetTest): ...@@ -122,7 +122,7 @@ class AIGradingWorkflowTest(CacheResetTest):
# Create a rubric with a similar structure, but different prompt # Create a rubric with a similar structure, but different prompt
similar_rubric_dict = copy.deepcopy(RUBRIC) similar_rubric_dict = copy.deepcopy(RUBRIC)
similar_rubric_dict['prompt'] = 'Different prompt!' similar_rubric_dict['prompts'] = [{"description": 'Different prompt!'}]
self.similar_rubric = rubric_from_dict(similar_rubric_dict) self.similar_rubric = rubric_from_dict(similar_rubric_dict)
def test_assign_most_recent_classifier_set(self): def test_assign_most_recent_classifier_set(self):
......
...@@ -161,7 +161,7 @@ class RubricHashTest(CacheResetTest): ...@@ -161,7 +161,7 @@ class RubricHashTest(CacheResetTest):
# Same structure, but different text should have the same structure hash # Same structure, but different text should have the same structure hash
altered_rubric = copy.deepcopy(RUBRIC) altered_rubric = copy.deepcopy(RUBRIC)
altered_rubric['prompt'] = 'altered!' altered_rubric['prompts'] = [{"description": 'altered!'}]
for criterion in altered_rubric['criteria']: for criterion in altered_rubric['criteria']:
criterion['prompt'] = 'altered!' criterion['prompt'] = 'altered!'
for option in criterion['options']: for option in criterion['options']:
......
...@@ -45,7 +45,7 @@ class Command(BaseCommand): ...@@ -45,7 +45,7 @@ class Command(BaseCommand):
] ]
RUBRIC = { RUBRIC = {
'prompt': u"Test prompt", 'prompts': [{"description": u"Test prompt"}],
'criteria': [ 'criteria': [
{ {
"order_num": 0, "order_num": 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