[Bug]: Unable to load dataset 'dell-research-harvard/newswire'

[JamesGuthrie]

What happened?

Attempting to load the above dataset fails:

postgres=# SELECT ai.load_dataset('dell-research-harvard/newswire');
ERROR:  AttributeError: 'NoneType' object has no attribute 'flatten'
CONTEXT:  Traceback (most recent call last):
  PL/Python function "load_dataset", line 35, in <module>
    return ai.load_dataset.load_dataset(
  PL/Python function "load_dataset", line 206, in load_dataset
  PL/Python function "load_dataset", line 47, in get_column_info
PL/Python function "load_dataset"```

### pgai extension affected

0.8.0

### pgai library affected

-

### PostgreSQL version used

17.2

### What operating system did you use?

Linux

### What installation method did you use?

Docker

### What platform did you run on?

On prem/Self-hosted

### Relevant log output and stack trace

```bash

How can we reproduce the bug?

SELECT ai.load_dataset('dell-research-harvard/newswire');

Are you going to work on the bugfix?

None