lots of changes!

Cargo.toml

@@ -10,7 +10,6 @@ crate-type = ["cdylib"]
 [dependencies]
 pyo3 = { version = "0.20.2", features = ["extension-module", "abi3-py38"] }
 numpy = "0.20.0"
-itertools = "0.12.0"
 neofoodclub = { path = "./neofoodclub_rs" }
 
 [profile.release]

neofoodclub/__init__.py

@@ -10,10 +10,8 @@
 from .models import *
 from .modifier import *
 from .neofoodclub import *
-from .nfc import *
 from .odds import *
 from .odds_change import *
-from .pirates import *
 
 __all__ = (
     "Math",

neofoodclub/arenas.py

@@ -2,9 +2,7 @@
 
 from typing import TYPE_CHECKING, Iterator, Sequence
 
-from neofoodclub.neofoodclub import Math
-
-from .pirates import Pirate
+from neofoodclub.neofoodclub import Math, Pirate
 
 if TYPE_CHECKING:
     from .nfc import NeoFoodClub

neofoodclub/bets.py

@@ -14,7 +14,7 @@
 if TYPE_CHECKING:
     import numpy.typing as npt
 
-    from neofoodclub.nfc import NeoFoodClub
+    from neofoodclub import NeoFoodClub
 
 __all__ = ("Bets",)
 
@@ -87,7 +87,8 @@ def bet_amounts(self) -> npt.NDArray[np.int32]:
 
     @bet_amounts.setter
     def bet_amounts(
-        self, val: Sequence[int | None] | npt.NDArray[np.int32] | None,
+        self,
+        val: Sequence[int | None] | npt.NDArray[np.int32] | None,
     ) -> None:
         if val is None:
             self._bet_amounts: npt.NDArray[np.int32] = np.array(
@@ -144,7 +145,10 @@ def __repr__(self) -> str:
 
     @classmethod
     def _from_generator(
-        cls, *, indices: npt.NDArray[np.int16], nfc: NeoFoodClub,
+        cls,
+        *,
+        indices: npt.NDArray[np.int16],
+        nfc: NeoFoodClub,
     ) -> Bets:
         # here is where we will take indices and sort as needed
         # to avoid confusion with "manually" making bets

neofoodclub/models/multinomial_logit.py

@@ -6,7 +6,7 @@
 from neofoodclub.models import BaseModel
 
 if TYPE_CHECKING:
-    from neofoodclub.nfc import NeoFoodClub
+    from neofoodclub import NeoFoodClub
 
 __all__ = [
     "MultinomialLogitModel",

neofoodclub/models/original.py

@@ -6,7 +6,7 @@
 from neofoodclub.neofoodclub import make_probabilities
 
 if TYPE_CHECKING:
-    from neofoodclub.nfc import NeoFoodClub
+    from neofoodclub import NeoFoodClub
 
 
 class OriginalModel(BaseModel):

neofoodclub/modifier.py

@@ -4,7 +4,7 @@
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from neofoodclub.nfc import NeoFoodClub
+    from neofoodclub import NeoFoodClub
 
 __all__ = ("Modifier",)