Improve type hints

[Armavica]

Description

Related Issue

• Closes #
• Related to #

Checklist

• Checked that the pre-commit linting/style checks pass
• Included tests that prove the fix is effective or that the new feature works
• Added necessary documentation (docstrings and/or example notebooks)
• If you are a pro: each commit corresponds to a relevant logical change

Type of change

• New feature / enhancement
• Bug fix
• Documentation
• Maintenance
• Other (please specify):

---

📚 Documentation preview 📚: https://pymc--7532.org.readthedocs.build/en/7532/

[codecov]

Codecov Report

Attention: Patch coverage is 96.42857% with 1 line in your changes missing coverage. Please review.

Project coverage is 92.76%. Comparing base (938aff4) to head (b98c47e).
Report is 2 commits behind head on main.

Files with missing lines	Patch %	Lines
pymc/backends/__init__.py	50.00%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #7532      +/-   ##
==========================================
+ Coverage   92.69%   92.76%   +0.06%     
==========================================
  Files         104      104              
  Lines       17409    17403       -6     
==========================================
+ Hits        16138    16144       +6     
+ Misses       1271     1259      -12     

Files with missing lines	Coverage Δ
pymc/backends/arviz.py	95.76% <ø> (ø)
pymc/backends/base.py	88.44% <ø> (ø)
pymc/distributions/custom.py	95.54% <ø> (ø)
pymc/distributions/distribution.py	91.89% <100.00%> (ø)
pymc/gp/hsgp_approx.py	88.39% <ø> (ø)
pymc/initial_point.py	99.09% <100.00%> (+0.05%)	⬆️
pymc/logprob/mixture.py	95.70% <100.00%> (ø)
pymc/model/fgraph.py	97.94% <ø> (ø)
pymc/sampling/forward.py	96.15% <ø> (ø)
pymc/sampling/jax.py	94.78% <ø> (ø)
... and 3 more