code: def test_aamped_A_B_join_KNN(T_A, T_B, dask_cluster):
      with Client(dask_cluster) as dask_client:
          m = 3
          for k in range(2, 4):
              for p in [1.0, 2.0, 3.0]:
                  ref_mp = naive.aamp(T_A, m, T_B=T_B, p=p, k=k)
                  comp_mp = aamped(
                      dask_client,
                      T_A,
                      m,
                      T_B,
                      ignore_trivial=False,
                      p=p,
                      k=k,
                  )
                  naive.replace_inf(ref_mp)
                  naive.replace_inf(comp_mp)
                  npt.assert_almost_equal(ref_mp, comp_mp)
  
imports: import naive
  import numpy
  import numpy.testing
  import pandas
  import pytest
  from dask.distributed import Client, LocalCluster
  from stumpy import aamped, config
label_desc: Use image node_id 31 for calling a global function with example usage: test_aamped_A_B_join_KNN(T_A, T_B, dask_cluster) without return types