From 2508fb57110738889800cfd82bcf8ce508e74971 Mon Sep 17 00:00:00 2001
From: Benjamin Datko <ben.datko@gmail.com>
Date: Tue, 2 Jan 2024 23:39:23 -0500
Subject: [PATCH] MVC to fix edge case for BoxModel and DeckofCards

See https://github.com/dlsun/symbulate/issues/93 for more details
---
 symbulate/probability_space.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/symbulate/probability_space.py b/symbulate/probability_space.py
index c03617c..81f8885 100644
--- a/symbulate/probability_space.py
+++ b/symbulate/probability_space.py
@@ -129,7 +129,7 @@ class BoxModel(ProbabilitySpace):
         sorted before returning or not.
     """
 
-    def __init__(self, box, size=None, replace=True, probs=None, order_matters=True):
+    def __init__(self, box, size=1, replace=True, probs=None, order_matters=True):
         if isinstance(box, list):
             self.box = box
             self.probs = probs
@@ -142,7 +142,7 @@ def __init__(self, box, size=None, replace=True, probs=None, order_matters=True)
             raise Exception(
                 "Box must be specified either as a list or a dict."
             )
-        self.size = None if size == 1 else size
+        self.size = size
         self.replace = replace
         self.order_matters = order_matters
         self.output_type = Vector
@@ -155,6 +155,7 @@ def __init__(self, box, size=None, replace=True, probs=None, order_matters=True)
                 "Cannot draw more tickets (without replacement) "
                 "than there are tickets in the box."
             )
+        self.size = None if size == 1 else size
 
     def draw(self):
         """