I am trying to replicate the following example of classification with Random Forests that is provided in the documentation: Classification - YDF documentation

There is an error in the last line of code in the example:

model.predict_class(test_ds)

**---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
Cell In[26], line 1
----> 1 model.predict_class(test_ds)

AttributeError: 'RandomForestModel' object has no attribute 'predict_class'**