Unsupported argument "target_object_key_format" in resource aws_s3_bucket_logging despite using aws-provider v5.31

Hi all, I’m trying to use the relatively new partitioned prefix functionality for server access logging described in Terraform Registry, however I’m getting the following error:

Error: Unsupported argument
│ on main.tf line 209, in resource "aws_s3_bucket_logging" "example":
│ 209: target_object_key_format = {
│ An argument named "target_object_key_format" is not expected here. Did you
│ mean to define a block of type "target_object_key_format"?
2024-01-18T19:47:47.127Z [DEBUG] provider.stdio: received EOF, stopping recv loop: err="rpc error: code = Unavailable desc = transport is closing"
2024-01-18T19:47:47.138Z [DEBUG] provider: plugin process exited: path=.terraform/providers/registry.terraform.io/hashicorp/aws/5.31.0/linux_amd64/terraform-provider-aws_v5.31.0_x5 pid=367
2024-01-18T19:47:47.138Z [DEBUG] provider: plugin exited

My resource configuration is as follows:

resource "aws_s3_bucket_logging" "example" {
  bucket = "bucket-1"

  target_bucket = "bucket-2"
  target_prefix = "log/"

  target_object_key_format = {
    partitioned_prefix = {
      partition_date_source = "DeliveryTime"

Has anyone seen this / knows a workaround?