From c414e90c28acfa11c44d7d967a88f9a407a72374 Mon Sep 17 00:00:00 2001
From: Vincent Bernat <vincent@bernat.ch>
Date: Mon, 12 Aug 2024 00:22:15 +0200
Subject: [PATCH] wallpaper: compute weights directly

---
 bin/wallpaper | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/bin/wallpaper b/bin/wallpaper
index ca5ff41..e42df54 100755
--- a/bin/wallpaper
+++ b/bin/wallpaper
@@ -218,8 +218,7 @@ def get_random_images(directory: str, number: int) -> list[Image]:
                     except (OSError, ValueError):
                         count = 0
                     counts.append(count)
-    counts = [(count - min(counts) + 1) ** 3 for count in counts]
-    weights = [100 / count for count in counts]
+    weights = [100 / ((count - min(counts) + 1) ** 3) for count in counts]
     images = [
         PIL.Image.open(image)
         for image in random.choices(image_files, k=number, weights=weights)