From 7fa18de3d3fcedfbac4c6e69cbe5292a932c4ba4 Mon Sep 17 00:00:00 2001
From: Hironsan <hiroki.nakayama.py@gmail.com>
Date: Wed, 1 Aug 2018 18:24:59 +0900
Subject: [PATCH] Update project model

---
 app/server/models.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/app/server/models.py b/app/server/models.py
index fd55b6e5..2e5085b2 100644
--- a/app/server/models.py
+++ b/app/server/models.py
@@ -4,7 +4,7 @@ from django.db import models
 from django.contrib.auth.models import User
 from django.contrib.staticfiles.storage import staticfiles_storage
 
-from .serializers import *
+
 
 
 class Project(models.Model):
@@ -66,6 +66,9 @@ class Project(models.Model):
         return docs
 
     def get_project_serializer(self):
+        from .serializers import DocumentSerializer
+        from .serializers import SequenceSerializer
+        from .serializers import Seq2seqSerializer
         if self.is_type_of(Project.DOCUMENT_CLASSIFICATION):
             return DocumentSerializer
         elif self.is_type_of(Project.SEQUENCE_LABELING):
@@ -76,6 +79,9 @@ class Project(models.Model):
             raise ValueError('Invalid project_type')
 
     def get_annotation_serializer(self):
+        from .serializers import DocumentAnnotationSerializer
+        from .serializers import SequenceAnnotationSerializer
+        from .serializers import Seq2seqAnnotationSerializer
         if self.is_type_of(Project.DOCUMENT_CLASSIFICATION):
             return DocumentAnnotationSerializer
         elif self.is_type_of(Project.SEQUENCE_LABELING):