Update utils.py

[aloctavodia]

---

📚 Documentation preview 📚: https://arviz-plots--171.org.readthedocs.build/en/171/

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 76.28%. Comparing base (5a7df08) to head (ee92651).

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #171      +/-   ##
==========================================
- Coverage   76.28%   76.28%   -0.01%     
==========================================
  Files          32       32              
  Lines        3741     3740       -1     
==========================================
- Hits         2854     2853       -1     
  Misses        887      887              

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

🚀 New features to boost your workflow:

• ❄ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.