When load saved policy model with flight_policy = tf.saved_model.load(flight_policy_path) in fly-env-examples.ipynb, it occur error below.
ValueError: The type 'tensorflow_probability.python.distributions.independent.Independent_ACTTypeSpec' has not been registered. It must be registered before you load this object (typically by importing its module).
Environment:
- Python 3.10
- tensorflow 2.8.0
- tensorflow-probability 0.16.0