feat: Add basic support to bool features

argilla/src/argilla/settings/_io/_hub.py

@@ -50,6 +50,7 @@ class FeatureType(Enum):
     LABEL = "label"
     INT = "int"
     FLOAT = "float"
+    BOOL = "bool"
 
 
 class AttributeType(Enum):
@@ -117,6 +118,8 @@ def _map_feature_type(feature):
             return FeatureType.INT
         elif dtype in ["float32", "float64"]:
             return FeatureType.FLOAT
+        elif dtype == "bool":
+            return FeatureType.BOOL
     elif hf_type == "Image":
         return FeatureType.IMAGE
     elif hf_type == "ClassLabel":
@@ -260,6 +263,8 @@ def _define_settings_from_features(
             metadata.append(IntegerMetadataProperty(name=name))
         elif feature_type == FeatureType.FLOAT:
             metadata.append(FloatMetadataProperty(name=name))
+        elif feature_type == FeatureType.BOOL:
+            metadata.append(TermsMetadataProperty(name=name))
         else:
             warnings.warn(f"Feature '{name}' has an unsupported type. Skipping. Feature type: {feature_type}")
 

argilla/tests/unit/test_settings/test_settings_from_features.py

@@ -56,6 +56,15 @@ def test_define_settings_from_features_with_non_curated_column_name(column: str,
     assert settings.fields[0].name == name
 
 
+def test_define_settings_from_bool_features():
+    features = {"text": {"_type": "Value", "dtype": "string"}, "bool_column": {"_type": "Value", "dtype": "bool"}}
+    settings = _define_settings_from_features(features, feature_mapping={})
+
+    assert len(settings.metadata) == 1
+    assert isinstance(settings.metadata[0], rg.TermsMetadataProperty)
+    assert settings.metadata[0].name == "bool_column"
+
+
 @pytest.mark.parametrize("column", ["text<column", "text>column", "text|column", "text]column", "text/column"])
 def test_define_settings_from_features_with_unsupported_column_name(column: str):
     features = {column: {"_type": "Value", "dtype": "string"}}