code: def _discover_pyx():
      exts = dict()
      for root, _, files in os.walk(os.path.join(repo_root, "mars")):
          for fn in files:
              if not fn.endswith(".pyx"):
                  continue
              full_fn = os.path.relpath(
                  os.path.join(root, fn), repo_root
              )
              include_dirs, source = ext_include_source_map.get(
                  full_fn.replace(os.path.sep, "/"), [[], []]
              )
              mod_name = full_fn.replace(".pyx", "").replace(
                  os.path.sep, "."
              )
              exts[mod_name] = Extension(
                  mod_name,
                  [full_fn] + source,
                  include_dirs=[np.get_include()] + include_dirs,
                  **cy_extension_kw,
              )
      return exts
  
imports: import os
  import platform
  import shutil
  import subprocess
  import sys
  import warnings
  from sysconfig import get_config_vars
  import numpy
  from Cython.Build import cythonize
  from pkg_resources import parse_version
  from setuptools import Command, Extension, setup
  from setuptools.command.develop import develop
  from setuptools.command.install import install
  from setuptools.command.sdist import sdist
label_desc: Use image node_id 1 for calling a global function with example usage: _discover_pyx() and returns: exts