How can I set hidden_units to a list in Vertex AI?

Question

I am following this notebook '02 ML Experimentation with Custom Model'.

When i try vertex_ai.log_params(hyperparams), I get:

TypeError: Value for key hidden_units is of type list but must be one of float, int, str

but the next step classifier = trainer.train needs hidden_units to be a list.

(My version of google-cloud-aiplatform is 1.16.0.) Any help is appreciated.