Error of running Merlin example on Google cloud platform

Hi, I am trying to run the example notebook 03-model-inference-hugectr.ipynb of Merlin on vertex AI.

My env is :

google cloud platform, Vertex AI, cloud shell

I am running the notebook from cloud shell. But I got an error when I run

export.export_ensemble(
 model_name=MODEL_NAME,
 workflow_path=local_workflow_path,
 saved_model_path=local_saved_model_path,
 output_path=local_ensemble_path,
 categorical_columns=categorical_columns,
 continuous_columns=continuous_columns,
 label_columns=label_columns,
 num_slots=NUM_SLOTS,
 max_nnz=MAX_NNZ,
 num_outputs=NUM_OUTPUTS,
 embedding_vector_size=EMBEDDING_VECTOR_SIZE,
 max_batch_size=MAX_BATCH_SIZE,
 model_repository_path=model_repository_path
)

The error:

   File '/home/.local/lib/python3.9/site-packages/merlin/dag/selector.py' line 151, in __bool__
   Attribute Error: 'ColumnSelector' object has no attribute 'all'

It is the file https://github.com/NVIDIA-Merlin/core/blob/main/merlin/dag/selector.py

Could anyone let me know how to work around it ?

thanks

[rnyak]

@umusa are you using docker image? if yes which one?