bertuner 0.2.0__tar.gz → 0.2.1__tar.gz

{bertuner-0.2.0 → bertuner-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bertuner
-Version: 0.2.0
+Version: 0.2.1
 Summary: Hyperparameter optimization and fine-tuning for BERT-style text classifiers (Optuna + MLflow), with long-context ModernBERT support
 Author-email: elemets <alafunnell@gmail.com>
 License: MIT

{bertuner-0.2.0 → bertuner-0.2.1}/bertuner/BERTuner.py

@@ -112,6 +112,17 @@ class BERTuneClassifier:
         self.target_cols = target_cols
         self.seed = seed
         self.df = pd.read_csv(data_path) if data_path is not None else dataframe.copy()
+        # Missing/NaN text values are coerced to empty strings so tokenization
+        # does not error on non-string inputs. Warn so callers know rows were
+        # altered rather than dropped.
+        n_missing = int(self.df[text_feature].isna().sum())
+        if n_missing:
+            warnings.warn(
+                f"{n_missing} missing (NaN) value(s) in text column "
+                f"'{text_feature}' were replaced with empty strings.",
+                stacklevel=2,
+            )
+        self.df[text_feature] = self.df[text_feature].fillna("").astype(str)
         if num_labels is not None:
             self.num_labels = num_labels
         elif self.is_multilabel:

{bertuner-0.2.0 → bertuner-0.2.1}/bertuner/__init__.py

@@ -1,6 +1,6 @@
 """bertuner: hyperparameter optimization and fine-tuning for BERT-style text classifiers."""
 
-__version__ = "0.2.0"
+__version__ = "0.2.1"
 
 from bertuner.BERTuner import BERTuneClassifier
 from bertuner.Predictor import BERTunePredictor

{bertuner-0.2.0 → bertuner-0.2.1}/bertuner.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bertuner
-Version: 0.2.0
+Version: 0.2.1
 Summary: Hyperparameter optimization and fine-tuning for BERT-style text classifiers (Optuna + MLflow), with long-context ModernBERT support
 Author-email: elemets <alafunnell@gmail.com>
 License: MIT

{bertuner-0.2.0 → bertuner-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "bertuner"
-version = "0.2.0"
+version = "0.2.1"
 description = "Hyperparameter optimization and fine-tuning for BERT-style text classifiers (Optuna + MLflow), with long-context ModernBERT support"
 readme = "README.md"
 license = { text = "MIT" }