We have
ext_modules=[ Extension( 'leap.learn.data_tools.file_io.extio', sources=['leap/learn/data_tools/file_io/extio.cpp'], depends=['leap/learn/data_tools/file_io/samples.h'], define_macros=[('NPY_NO_DEPRECATED_API', 'NPY_1_9_API_VERSION')], extra_compile_args=['-std=c++11'], libraries=['rt'] if platform.system() == 'Linux' else [], include_dirs=[GetNumpyIncludeDirectoryLazy()], optional=True ),in our setup.py which I believe isn't being built correctly when the task is running on the agent.

Manually I was installing the leap package through python -m pip install . when building the docker container. My thinking was that when the tasks environment was then replicated on the agent, the leap package would be installed correctly through it's setup.py with the Extension which I've listed above