[AIR][DOC] Improved code snippet from TorchCheckpoint, removed angle brackets, and assert outputs

[dmatrix]

Signed-off-by: Jules Damji [email protected]

• As part of on-going effort to make code snippets runnable
• Use testcode and testoput for CI
• Assert output

Checks

• [x ] I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
• [ x] I've run scripts/format.sh to lint the changes in this PR.

[pcmoritz]

Thanks, looks great! There is still a little convincing of the CI to do I think before we can merge it :)

[pcmoritz]

Is it possible that the CI wants you to do

            .. testoutput::
                :hide:
                :options: +ELLIPSIS

instead of

            .. testoutput::
            :hide:
            :options: +ELLIPSIS

?

[dmatrix]

@pcmoritz a bit weird that all local doctest pass, but somehow the ci-build-pr buildkite fails.... adding and removing the same lines ??