Open umusa opened 1 year ago
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
@umusa are you using docker image? if yes which one?
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
The error:
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