Library fix and annotation fix

[neeleshverma]

numpy should be used instead of math for better precision and a small annotation fix.

[Madhav-MKNC]

Changes:

• Replacing math.prod with np.prod is acceptable if the numpy package is imported.
• Addition of params_only: bool parameter requires updates to function calls and documentation.

Recommendations:

• Ensure unit tests are updated for the new changes.
• Confirm the changes align with the system's design and do not introduce regressions.

If the above points are addressed, proceed with merging the PR after successful CI checks.

[neeleshverma]

The "params_only" parameter is already there in the function. I simply added the type (bool) to make the function declaration clear.