Module Not Found: yggdrasil_decision_forests.model.gradient_boosted_tree
@ngostin,
I tried to import the tensorflow_decision_forests in the latest version Tensorflow 2.16.1 and could not face any error/issue while importing. Kindly find the gist of it [here](https://colab.research.google.com/gist/tilakrayal/f42d057225455cc7203570d65bdeb58c/untitled1934.ipynb).
```python
!pip install tensorflow-decision-forests
```

Also there is another repo for the TensorFlow decision forests for the quick resolution. Please try to post in that repo.
https://github.com/tensorflow/decision-forests/issues

Thank you!
### Issue type

Bug

### Have you reproduced the bug with TensorFlow Nightly?

No

### Source

source

### TensorFlow version

2.16.1

### Custom code

Yes

### OS platform and distribution

Mac OS Big Sur 11.3.1

### Mobile device

_No response_

### Python version

3.9

### Bazel version

_No response_

### GCC/compiler version

_No response_

### CUDA/cuDNN version

_No response_

### GPU model and memory

_No response_

### Current behavior?

I've been wrestling trying to install/implement some form of Decision Forests for weeks now, and keep running into issues with Both ydf and tf-df. Both issues seem to happen on import. For tf-df, I get a module not found error, relating to yggdrasil_decision_forests

(less related, the issue I'm having with ydf gives an error "no suitable image found" for ydf)

### Standalone code to reproduce the issue

```shell
### import required packages, define constants

import tensorflow_decision_forests as tfdf
```


### Relevant log output

```shell
2024-06-04 19:22:02.418757: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[1], line 10
      7 import pandas as pd
      9 #import ydf  # Yggdrasil Decision Forests
---> 10 import tensorflow_decision_forests as tfdf
     12 pi = np.pi

File ~/opt/anaconda3/lib/python3.9/site-packages/tensorflow_decision_forests/__init__.py:64
     60 from tensorflow_decision_forests.tensorflow import check_version
     62 check_version.check_version(__version__, compatible_tf_versions)
---> 64 from tensorflow_decision_forests import keras
     65 from tensorflow_decision_forests.component import py_tree
     66 from tensorflow_decision_forests.component.builder import builder

File ~/opt/anaconda3/lib/python3.9/site-packages/tensorflow_decision_forests/keras/__init__.py:53
     15 """Decision Forest in a Keras Model.
     16 
     17 Usage example:
   (...)
     48 
     49 """
     51 from typing import Callable, List
---> 53 from tensorflow_decision_forests.keras import core
     54 from tensorflow_decision_forests.keras import wrappers
     56 # Utility classes

File ~/opt/anaconda3/lib/python3.9/site-packages/tensorflow_decision_forests/keras/core.py:62
     60 from tensorflow.python.data.ops import dataset_ops
     61 from tensorflow.python.data.ops import load_op
---> 62 from tensorflow_decision_forests.component.inspector import inspector as inspector_lib
     63 from tensorflow_decision_forests.component.tuner import tuner as tuner_lib
     64 from tensorflow_decision_forests.keras import core_inference

File ~/opt/anaconda3/lib/python3.9/site-packages/tensorflow_decision_forests/component/inspector/inspector.py:71
     69 from yggdrasil_decision_forests.model import hyperparameter_pb2
     70 from yggdrasil_decision_forests.model.decision_tree import decision_tree_pb2
---> 71 from yggdrasil_decision_forests.model.gradient_boosted_trees import gradient_boosted_trees_pb2
     72 from yggdrasil_decision_forests.model.multitasker import multitasker_pb2
     73 from yggdrasil_decision_forests.model.random_forest import random_forest_pb2

ModuleNotFoundError: No module named 'yggdrasil_decision_forests.model.gradient_boosted_trees'
```
