Unable to import tf_quant_finance as tff in Colab notebook: Monte_Carlo_Euler_Scheme.ipynb

[ay1011]

import matplotlib.pyplot as plt import numpy as np import time

import tensorflow as tf

import QuantLib as ql

tff for Tensorflow Finance

import tf_quant_finance as tff

from IPython.core.pylabtools import figsize figsize(21, 14) # better graph size for Colab

---

ValueError Traceback (most recent call last) in <cell line: 12>() 10 11 # tff for Tensorflow Finance ---> 12 import tf_quant_finance as tff 13 14 from IPython.core.pylabtools import figsize

16 frames /usr/local/lib/python3.10/dist-packages/tensorflow_probability/python/internal/prefer_static.py in _copy_docstring(original_fn, new_fn) 82 new_spec = tf_inspect.getfullargspec(new_fn) 83 if original_spec != new_spec: ---> 84 raise ValueError( 85 'Arg specs do not match: original={}, new={}, fn={}'.format( 86 original_spec, new_spec, original_fn))

ValueError: Arg specs do not match: original=FullArgSpec(args=['input', 'dtype', 'name', 'layout'], varargs=None, varkw=None, defaults=(None, None, None), kwonlyargs=[], kwonlydefaults=None, annotations={}), new=FullArgSpec(args=['input', 'dtype', 'name'], varargs=None, varkw=None, defaults=(None, None), kwonlyargs=[], kwonlydefaults=None, annotations={}), fn=<function ones_like_v2 at 0x7d8479b6c5e0>