When doing `flyte.wait` in flyte-sdk, it will call `WatchActionDetails`. Now it does not work now and we should make it work