fix(index): allow date32 ordered scalar indexes

lance_ray/index.py

@@ -588,12 +588,13 @@ def create_scalar_index(
                 is_supported = (
                     pa.types.is_integer(value_type)
                     or pa.types.is_floating(value_type)
+                    or pa.types.is_date32(value_type)
                     or pa.types.is_string(value_type)
                     or pa.types.is_large_string(value_type)
                 )
                 if not is_supported:
                     raise TypeError(
-                        f"Column {column} must be numeric or string type for "
+                        f"Column {column} must be numeric, date32, or string type for "
                         f"{index_type} index, got {value_type}"
                     )
             case "LABEL_LIST":

tests/test_vector_index_options.py

@@ -69,7 +69,7 @@ def id(self):
 
 class _FakeLanceSchema:
     def field(self, column):
-        if column not in {"value", "text", "labels"}:
+        if column not in {"value", "text", "labels", "event_date"}:
             raise KeyError(column)
         return _FakeLanceField()
 
@@ -84,6 +84,8 @@ def field(self, column):
             return _FakeField(column, index_mod.pa.string())
         if column == "labels":
             return _FakeField(column, index_mod.pa.list_(index_mod.pa.string()))
+        if column == "event_date":
+            return _FakeField(column, index_mod.pa.date32())
         else:
             raise KeyError(column)
 
@@ -94,6 +96,7 @@ def __iter__(self):
                 _FakeField("value", index_mod.pa.int64()),
                 _FakeField("text", index_mod.pa.string()),
                 _FakeField("labels", index_mod.pa.list_(index_mod.pa.string())),
+                _FakeField("event_date", index_mod.pa.date32()),
             ]
         )
 
@@ -452,13 +455,15 @@ def test_create_index_rejects_invalid_num_segments(monkeypatch):
     ("index_type", "column"),
     [
         ("BTREE", "value"),
+        ("BTREE", "event_date"),
         ("BITMAP", "value"),
         ("INVERTED", "text"),
         ("FTS", "text"),
         ("NGRAM", "text"),
         ("BLOOMFILTER", "value"),
         ("RTREE", "value"),
         ("LABEL_LIST", "labels"),
+        ("ZONEMAP", "event_date"),
     ],
 )
 def test_create_scalar_index_uses_segment_path(monkeypatch, index_type, column):