oh-my-batch 0.4.3__tar.gz → 0.4.5__tar.gz

{oh_my_batch-0.4.3 → oh_my_batch-0.4.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: oh-my-batch
-Version: 0.4.3
+Version: 0.4.5
 Summary: 
 License: GPL
 Author: weihong.xu
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: fire (>=0.7.0,<0.8.0)
+Requires-Dist: fire (>=0.5.0,<0.6.0)
 Description-Content-Type: text/markdown
 
 # oh-my-batch

{oh_my_batch-0.4.3 → oh_my_batch-0.4.5}/oh_my_batch/combo.py

@@ -33,7 +33,7 @@ class ComboMaker:
         self.add_var(key, *args)
         return self
 
-    def add_randint(self, key: str, n: int, a: int, b: int, seed=None):
+    def add_randint(self, key: str, n: int, a: int, b: int, uniq=False, seed=None):
         """
         Add a variable with random integer values
 
@@ -41,11 +41,17 @@ class ComboMaker:
         :param n: Number of values
         :param a: Lower bound
         :param b: Upper bound
+        :param uniq: If True, values are unique, default is False
         :param seed: Seed for random number generator
         """
         if seed is not None:
             random.seed(seed)
-        args = [random.randint(a, b) for _ in range(n)]
+        if uniq:
+            if b - a + 1 < n:
+                raise ValueError("Not enough unique values")
+            args = random.sample(range(a, b + 1), n)
+        else:
+            args = [random.randint(a, b) for _ in range(n)]
         self.add_var(key, *args)
         return self
 

{oh_my_batch-0.4.3 → oh_my_batch-0.4.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "oh-my-batch"
-version = "0.4.3"
+version = "0.4.5"
 description = ""
 authors = ["weihong.xu <xuweihong.cn@gmail.com>"]
 license = "GPL"
@@ -9,7 +9,7 @@ packages = [{include = "oh_my_batch"}]
 
 [tool.poetry.dependencies]
 python = "^3.8"
-fire = "^0.7.0"
+fire = "^0.5.0"
 
 [build-system]
 requires = ["poetry-core"]