From d692337b9af166fbe75f709775fa7562c45e9652 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Mon, 17 May 2021 11:45:11 +0900
Subject: [PATCH] Update filter option for image classification

---
 frontend/domain/models/project/project.ts | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/frontend/domain/models/project/project.ts b/frontend/domain/models/project/project.ts
index 5fb3c6ae..9eb572c2 100644
--- a/frontend/domain/models/project/project.ts
+++ b/frontend/domain/models/project/project.ts
@@ -91,11 +91,13 @@ export class ProjectReadItem {
 
   get filterOption() {
     if (this.project_type === 'DocumentClassification') {
-      return 'doc_annotations__isnull'
+      return 'categories__isnull'
     } else if (this.project_type === 'SequenceLabeling') {
-      return 'seq_annotations__isnull'
+      return 'spans__isnull'
     } else if (this.project_type === 'Seq2seq') {
-      return 'seq2seq_annotations__isnull'
+      return 'texts__isnull'
+    } else if (this.project_type === 'ImageClassification') {
+      return 'categories__isnull'
     } else {
       return ''
     }