Fix integration between staff info button and AI training API data formats
Merge pull request #352 from edx/will/ai-grading-test-fixup

Will/ai grading test fixup