fix: check for first number in status to determine if file is trained…

… or untrained
jakowenko · Sep 3, 2021 · 125f96e · 125f96e
1 parent 809d5f3
commit 125f96e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/frontend/src/views/Train.vue b/frontend/src/views/Train.vue
@@ -131,10 +131,12 @@ export default {
           try {
             const ids = $this.matches.selected.map((obj) => obj.id);
             const trained = $this.matches.selected.filter(
-              (obj) => obj.results.filter((res) => res.result.status === 200).length,
+              (obj) => obj.results.filter((res) => res.result.status.toString().charAt(0) === '2').length,
             );
             const untrained = $this.matches.selected.filter(
-              (obj) => !obj.results.length || obj.results.filter((res) => res.result.status !== 200).length,
+              (obj) =>
+                !obj.results.length ||
+                obj.results.filter((res) => res.result.status.toString().charAt(0) !== '2').length,
             );
             const names = [...new Set(trained.map((obj) => obj.name))];
             let message = '';