Add argument for code-block directive

To fix a build warning https://www.sphinx-doc.org/en/master/usage/restructuredtext/directives.html#directive-code-block