Update driver.py

[anonymousdouble]

refactor with chain assignment to make code more Pythonic, concise and efficient.

• Have you signed the contributor license agreement?
• Have you followed the contributor guidelines?
• Have you run make check-all successfully?
• Did you choose a descriptive title and description for your PR?
• (Only for maintainers) Did you apply appropriate labels and a milestone?

[cla-checker-service[bot]]

❌ Author of the following commits did not sign a Contributor Agreement: 7f2d403c84a03881985040adeeb7c3b39c6db21d

Please, read and sign the above mentioned agreement if you want to contribute to this project