Fix: Resolving of import paths for some torch functions not working

[mauvilsa]

What does this PR do?

Fixes #534

Before submitting

• Did you read the contributing guideline?
• [n/a] Did you update the documentation? (readme and public docstrings)
• Did you write unit tests such that there is 100% coverage on related code? (required for bug fixes and new features)
• Did you verify that new and existing tests pass locally?
• Did you make sure that all changes preserve backward compatibility?
• Did you update the CHANGELOG? (not for typos, docs, test updates, or minor internal changes/refactors)

[sonarcloud]

Quality Gate passed

Issues
0 New issues
0 Accepted issues

Measures
0 Security Hotspots
0.0% Coverage on New Code
0.0% Duplication on New Code

See analysis details on SonarCloud

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 100.00%. Comparing base (2bcbd48) to head (d4ac134).

Additional details and impacted files

@@            Coverage Diff            @@
##              main      #535   +/-   ##
=========================================
  Coverage   100.00%   100.00%           
=========================================
  Files           22        22           
  Lines         6374      6377    +3     
=========================================
+ Hits          6374      6377    +3     

Flag	Coverage Δ
py3.10	85.64% <100.00%> (+<0.01%)	⬆️
py3.10_all	98.68% <100.00%> (+<0.01%)	⬆️
py3.10_pydantic1	48.56% <20.00%> (-0.03%)	⬇️
py3.10_pydantic2	48.34% <20.00%> (-0.03%)	⬇️
py3.10_types	98.69% <100.00%> (+<0.01%)	⬆️
py3.11	85.63% <100.00%> (+<0.01%)	⬆️
py3.11_all	98.68% <100.00%> (+<0.01%)	⬆️
py3.11_types	98.69% <100.00%> (+<0.01%)	⬆️
py3.12	85.77% <100.00%> (+<0.01%)	⬆️
py3.12_all	98.68% <100.00%> (+<0.01%)	⬆️
py3.12_types	98.69% <100.00%> (+<0.01%)	⬆️
py3.7	86.12% <100.00%> (+<0.01%)	⬆️
py3.7_all	99.24% <100.00%> (+<0.01%)	⬆️
py3.7_types	99.27% <100.00%> (+<0.01%)	⬆️
py3.8	86.31% <100.00%> (+<0.01%)	⬆️
py3.8_all	99.40% <100.00%> (+<0.01%)	⬆️
py3.8_types	99.41% <100.00%> (+<0.01%)	⬆️
py3.9	86.17% <100.00%> (+<0.01%)	⬆️
py3.9_all	99.29% <100.00%> (+<0.01%)	⬆️
py3.9_types	99.30% <100.00%> (+<0.01%)	⬆️

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

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