Looks like TEDPolicy is failing to learn the next steps. Can you try to increase the epochs for TEDPolicy (try 300), and maybe try to replace MemoizationPolicy with AugmentedMemoizationPolicy?
It also looks like this structure might be suited for some rules. Have you considered adding some rules?