Brexit department warns EU counterparts it will ‘return waste to its country of origin’ if an agreement on nuclear cooperation cannot be reached