Add plugin jobs statuses admin
@@ -3,7 +3,8 @@ from __future__ import absolute_import, division, print_function
from django.contrib import admin
-from plugins.models import TensorboardJob, NotebookJob
+from jobs.admin import JobStatusAdmin
+from plugins.models import TensorboardJob, NotebookJob, TensorboardJobStatus, NotebookJobStatus
class TensorboardJobAdmin(admin.ModelAdmin):
@@ -14,5 +15,15 @@ class NotebookJobAdmin(admin.ModelAdmin):
readonly_fields = ('created_at', 'updated_at')
+class TensorboardJobStatusAdmin(JobStatusAdmin):
+ pass
+
+
+class NotebookJobStatusAdmin(JobStatusAdmin):
+ pass
+
+
admin.site.register(TensorboardJob, TensorboardJobAdmin)
admin.site.register(NotebookJob, NotebookJobAdmin)
+admin.site.register(TensorboardJobStatus, TensorboardJobStatusAdmin)
+admin.site.register(NotebookJobStatus, NotebookJobStatusAdmin)
