Better handling of imports errors when blingfire is not available (#61)

* adding extra errors

* style

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "smashed"
-version = "0.21.3"
+version = "0.21.4"
 description = """\
 SMASHED is a toolkit designed to apply transformations to samples in \
 datasets, such as fields extraction, tokenization, prompting, batching, \
@@ -11,7 +11,7 @@ license = {text = "Apache-2.0"}
 readme = "README.md"
 requires-python = ">=3.8"
 dependencies = [
-    "necessary>=0.4.1",
+    "necessary>=0.4.3",
     "trouting>=0.3.3",
     "ftfy>=6.1.1",
     "platformdirs>=2.5.0",

src/smashed/utils/wordsplitter.py

@@ -2,7 +2,9 @@
 
 from necessary import Necessary, necessary
 
-with necessary("blingfire", soft=True) as BLINGFIRE_AVAILABLE:
+with necessary(
+    "blingfire", soft=True, errors=(ModuleNotFoundError, OSError)
+) as BLINGFIRE_AVAILABLE:
     if BLINGFIRE_AVAILABLE or TYPE_CHECKING:
         from blingfire import text_to_words
 
@@ -39,6 +41,7 @@ def __call__(
 
 @Necessary(
     "blingfire",
+    errors=(ModuleNotFoundError, OSError),
     message=(
         "{module_name} missing. Fix with 'pip install smashed[prompting]'."
         "If you are on a Mac with Apple Silicon chip, also run "

tests/test_batch_interface.py

@@ -17,7 +17,7 @@ class TestBatchInterface(unittest.TestCase):
     def test_batch(self, remove_columns: bool = False):
         mapper = MockMapper(1, output_fields=["a"])
 
-        data = Dataset.from_list([{"a": i, "b": i**2} for i in range(100)])
+        data = Dataset.from_list([{"a": i, "b": i ** 2} for i in range(100)])
 
         def _batch_fn(data: LazyBatch, mapper: MockMapper) -> LazyBatch:
             return mapper.map(deepcopy(data), remove_columns=remove_columns)

tests/test_caching.py

@@ -37,7 +37,7 @@ def test_list_cache(self):
 
     def test_datasets_cache(self):
         dt = Dataset.from_dict(
-            {"a": [i for i in range(5)], "b": [i**2 for i in range(5)]}
+            {"a": [i for i in range(5)], "b": [i ** 2 for i in range(5)]}
         )
 
         with tempfile.TemporaryDirectory() as tmpdir: