Add AI models to the admin page

apps/openassessment/assessment/admin.py

@@ -5,7 +5,8 @@ from django.core.urlresolvers import reverse
 from django.utils import html
 
 from openassessment.assessment.models import (
-    Assessment, AssessmentFeedback, PeerWorkflow, PeerWorkflowItem, Rubric
+    Assessment, AssessmentFeedback, PeerWorkflow, PeerWorkflowItem, Rubric,
+    AIGradingWorkflow, AITrainingWorkflow, AIClassifierSet, AIClassifier
 )
 from openassessment.assessment.serializers import RubricSerializer
 
@@ -119,7 +120,33 @@ class AssessmentFeedbackAdmin(admin.ModelAdmin):
     assessments_by.allow_tags = True
 
 
+class AIGradingWorkflowAdmin(admin.ModelAdmin):
+    list_display = ('uuid', 'submission_uuid')
+    search_fields = ('uuid', 'submission_uuid', 'student_id', 'item_id', 'course_id')
+    readonly_fields = ('uuid', 'submission_uuid', 'student_id', 'item_id', 'course_id')
+
+
+class AITrainingWorkflowAdmin(admin.ModelAdmin):
+    list_display = ('uuid',)
+    # TODO -- update to include student/item/course id
+    search_fields = ('uuid',)
+    readonly_fields = ('uuid',)
+
+
+class AIClassifierInline(admin.TabularInline):
+    model = AIClassifier
+
+
+class AIClassifierSetAdmin(admin.ModelAdmin):
+    list_display = ('id',)
+    search_fields = ('id',)
+    inlines = [AIClassifierInline]
+
+
 admin.site.register(Rubric, RubricAdmin)
 admin.site.register(PeerWorkflow, PeerWorkflowAdmin)
 admin.site.register(Assessment, AssessmentAdmin)
 admin.site.register(AssessmentFeedback, AssessmentFeedbackAdmin)
+admin.site.register(AIGradingWorkflow, AIGradingWorkflowAdmin)
+admin.site.register(AITrainingWorkflow, AITrainingWorkflowAdmin)
+admin.site.register(AIClassifierSet, AIClassifierSetAdmin)