From 1da2d9fe9c91ed32b88cee46a3954ee71a458f77 Mon Sep 17 00:00:00 2001
From: Bel LaPointe <breel@qualtrics.com>
Date: Tue, 20 Sep 2022 15:04:37 -0600
Subject: [PATCH] one more

---
 src/bucket.py      | 4 ++++
 src/test_bucket.py | 1 +
 2 files changed, 5 insertions(+)

diff --git a/src/bucket.py b/src/bucket.py
index c4868b2..c0b25e7 100644
--- a/src/bucket.py
+++ b/src/bucket.py
@@ -54,3 +54,7 @@ class BucketChooserProportionalLogRandom:
             1 + int(100 * math.log2(i[1])),
          ) for i in content
       ]
+
+class BucketChooserProportionalRandom(BucketChooserProportionalLogRandom):
+   def xform(self, content):
+      return [(i[0], 1+int(100*i[1])) for i in content]
diff --git a/src/test_bucket.py b/src/test_bucket.py
index 93ecb9a..11ef34a 100644
--- a/src/test_bucket.py
+++ b/src/test_bucket.py
@@ -34,6 +34,7 @@ class TestBucketChoosers(unittest.TestCase):
    def test_choosers(self):
       for chooser in [
          bucket.BucketChooserProportionalLogRandom(),
+         bucket.BucketChooserProportionalRandom(),
       ]:
          with self.assertRaises(AssertionError):
             self.assertRaises(chooser.choose([]))