code: def test_pyx():
      pred_probs = np.array(
          [
              [0.1, 0.3, 0.6],
              [0.1, 0.0, 0.9],
              [0.1, 0.0, 0.9],
              [1.0, 0.0, 0.0],
              [0.1, 0.8, 0.1],
          ]
      )
      ps, py, inv = get_latent_py_ps_inv()
      pyx = latent_algebra.compute_pyx(pred_probs, nm, inv)
      assert np.all(np.sum(pyx, axis=1) - 1 < 1e-4)
  
file_path: cleanlab/tests/test_latent_algebra.py
imports: from cleanlab.internal import latent_algebra
  import numpy
  import pytest
label_desc: Use image node_id 9 for calling a global function with example usage: test_pyx() without return types