Open BrandonStudio opened 1 year ago
Hello, have you solved this issue? i also have the same error when initializing tea.seq2seq.LuongAttention.
Do you have a very minimal gist to run to reproduce this?
@We-here unfortunately no, I have not found a way to bypass type check
@bhack I think the code above is enough. Exception is thrown when initializing the instance of the class, not afterwards
Yes cause _process_probability_fn
was going to transform str
back to the function.
Can you send a PR to the specific test to help reproduce this case?
@seanpmorgan
Also seeing this issue on google colab.
@BrandonStudio I skip the type checking by comment out @typechecked
at AttentionMechanism
and its derived classes
@DangMinh24 Did you just modify the library source code?
@BrandonStudio Yeah, I modify tensorflow-addons code in local environment.
System information
Describe the bug
call BahdanauAttention with default
probability_fn
value ("softmax"
) raises type errorTried to debug and found that
probability_fn
was a function when checking typeCode to reproduce the issue
Other info / logs