Improves optional imports

[mattpopovich]

Fixes #353

[CLAassistant]

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution.
1 out of 2 committers have signed the CLA.

✅ mattpopovich
❌ pre-commit-ci[bot]
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

[codecov]

Codecov Report

Merging #366 (ea0291a) into main (3c14e91) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##             main     #366   +/-   ##
=======================================
  Coverage   98.58%   98.58%           
=======================================
  Files          11       11           
  Lines         778      778           
=======================================
  Hits          767      767           
  Misses         11       11           

Flag	Coverage Δ
unittests	98.58% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 3c14e91...ea0291a. Read the comment docs.

[zhiqwang]

Thanks @mattpopovich , the overall looks great to me! I only make one comment about where to place the decorator, let me know if you have more concerns.

[zhiqwang]

Hi @mattpopovich , since #369 introduces some conflicts, I rebase this PR manually, and please use the following command to synchronize with the current branch:

git pull --rebase origin improve-optional-imports

[zhiqwang]

This looks GREAT to me, thanks @mattpopovich for your great contributions here! And I'm merging this PR!