Allow checkpoint creation when partion column is "timestampNtz "

[thomasfrederikhoeck]

Description

Right now you can create a table partioned on a column of type timestampNtz. This is all on main.:

import pandas as pd
dates = pd.date_range(datetime(2021,1,1,3,4,6,3),datetime(2021,1,3,3,4,6))
df = pd.DataFrame({"time":dates, "a":[i for i in range(len(dates))]})
write_deltalake("mytable",df, partition_by="time")
dt = DeltaTable("mytable")
print(dt.schema())
> Schema([Field(time, PrimitiveType("timestampNtz"), nullable=True), Field(a, PrimitiveType("long"), nullable=True)])

However when trying to create a checkpoint I'm hit by

dt.create_checkpoint()
> PanicException: not implemented: Primitive type timestampNtz is not supported for partition column values.

Use Case

Related Issue(s)

[ion-elgreco]

@thomasfrederikhoeck We just need to add an arm for PrimitiveType::TimestampNtz`

delta-rs/crates/core/src/protocol/checkpoints.rs

Line 444 in 6f81b80

PrimitiveType::Timestamp => {

I'll see if I can find some time tomorrow

[thomasfrederikhoeck]

GIven #2380 it might also be worth looking at the Timestamp arm?

[thomasfrederikhoeck]

@ion-elgreco I can grap this one :-) I seems like you are already started :-D