Fix NotLuksus#23, NotLuksus#24

yrik · Sep 30, 2021 · 1568d18 · 1568d18
1 parent e8d8e51
commit 1568d18
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bin/index.js b/bin/index.js
@@ -366,7 +366,7 @@ async function generateImages() {
   let id = 0;
   await generateWeightedTraits();
   if (config.deleteDuplicates) {
-    while (weightedTraits[0].length > 0 && noMoreMatches < 20000) {
+    while (!Object.values(weightedTraits).filter(arr => arr.length == 0).length && noMoreMatches < 20000) {
       let picked = [];
       order.forEach(id => {
         let pickedImgId = pickRandom(weightedTraits[id]);
@@ -392,7 +392,7 @@ async function generateImages() {
       }
     }
   } else {
-    while (weightedTraits[0].length > 0) {
+    while (!Object.values(weightedTraits).filter(arr => arr.length == 0).length) {
       order.forEach(id => {
         images.push(
           basePath + traits[id] + '/' + pickRandomAndRemove(weightedTraits[id])