Fixed a bug where an error was raised where the item argument of __getitem__ or __setitem__ was a bool.

Author: roytsmart

named_arrays/_functions/functions.py

@@ -438,6 +438,9 @@ def _getitem(
         inputs = array.inputs
         outputs = array.outputs
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractArray):
             if isinstance(item, na.AbstractFunctionArray):
                 if not np.all(item.inputs == array.inputs):
@@ -991,6 +994,9 @@ def __setitem__(
             value: float | u.Quantity | na.FunctionArray,
     ):
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractFunctionArray):
             if not np.all(item.inputs == self.inputs):
                 raise ValueError("boolean advanced index does not have the same inputs as the array")

named_arrays/_scalars/scalars.py

@@ -393,6 +393,9 @@ def _getitem(
             item: dict[str, int | slice | AbstractScalarArray] | AbstractScalarArray,
     ):
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, AbstractScalarArray):
 
             if not set(item.shape).issubset(self.axes):
@@ -1016,6 +1019,9 @@ def __setitem__(
         else:
             value = ScalarArray(value)
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, AbstractScalarArray):
 
             item = item.explicit

named_arrays/_scalars/uncertainties/uncertainties.py

@@ -268,6 +268,9 @@ def _getitem(
         nominal = na.as_named_array(array.nominal)
         distribution = na.as_named_array(array.distribution)
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractArray):
             item = item.explicit
             if isinstance(item, AbstractUncertainScalarArray):
@@ -649,6 +652,9 @@ def __setitem__(
     ):
         shape_self = self.shape
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractArray):
 
             item = item.explicit

named_arrays/_vectors/vectors.py

@@ -262,6 +262,9 @@ def _getitem(
         shape_array = array.shape
         components = array.components
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractArray):
             item = item.explicit
             shape_item = item.shape
@@ -610,6 +613,9 @@ def __setitem__(
     ):
         components_self = self.components
 
+        if isinstance(item, bool):
+            item = na.as_named_array(item)
+
         if isinstance(item, na.AbstractArray):
             if isinstance(item, na.AbstractVectorArray):
                 if item.type_abstract == self.type_abstract: