Partly revert how SetFit seed bounds are set (#72)

Signed-off-by: Christopher Schröder <[email protected]>
webis-de · Jan 21, 2025 · 8adaeaa · 8adaeaa
1 parent bfd80e7
commit 8adaeaa
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 37 deletions.
diff --git a/small_text/integrations/transformers/classifiers/setfit.py b/small_text/integrations/transformers/classifiers/setfit.py
@@ -15,7 +15,6 @@
 )
 from small_text.utils.context import build_pbar_context
 from small_text.utils.labels import csr_to_list
-from small_text.utils.numpy import int_dtype
 from small_text.integrations.transformers.classifiers.base import (
     ModelLoadingStrategy
 )
@@ -258,7 +257,7 @@ def _get_train_and_valid_sets(self, x_train, y_train, x_valid, y_valid):
         return sub_train, sub_valid
 
     def _fit(self, sub_train, sub_valid, setfit_train_kwargs):
-        seed = np.random.randint(np.iinfo(int_dtype()).max)
+        seed = np.random.randint(np.iinfo(np.uint32).max, dtype=np.uint32).item()
         trainer = SetFitTrainer(
             self.model,
             sub_train,

diff --git a/small_text/utils/numpy.py b/small_text/utils/numpy.py
diff --git a/tests/unit/small_text/utils/test_numpy.py b/tests/unit/small_text/utils/test_numpy.py