Small fix for an overly aggressive warning

[austingmhuang]

Context:
Was raising a warning too aggressively.

Description of the Change:
Small change to make sure using numpy with autograd/jax doesn't raise a warning.

Benefits:

Possible Drawbacks:

Related GitHub Issues:

[github-actions]

Hello. You may have forgotten to update the changelog!
Please edit doc/releases/changelog-dev.md with:

• A one-to-two sentence description of the change. You may include a small working example for new features.
• A link back to this PR.
• Your name (or GitHub username) in the contributors section.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 99.62%. Comparing base (056dd9a) to head (041ec0e).
Report is 1 commits behind head on master.

Additional details and impacted files

@@           Coverage Diff           @@
##           master    #6707   +/-   ##
=======================================
  Coverage   99.62%   99.62%           
=======================================
  Files         476      476           
  Lines       44740    44740           
=======================================
  Hits        44571    44571           
  Misses        169      169           

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[soranjh]

Thanks @austingmhuang, looks good to go.