#5020 [BUG] Core dump when running kf-pytorch plugin tests on python 3.12 Issue created by eapolinario ### Describe the bug As the title says. Tests for the kf-pytorch plugin core dumps: ``` platform linux -- Python 3.12.2, pytest-8.0.2, pluggy-1.4.0 rootdir: /home/runner/work/flytekit/flytekit configfile: pyproject.toml plugins: mock-3.12.0, timeout-2.2.0, xdist-3.5.0, asyncio-0.23.5, cov-4.1.0, hypothesis-6.98.17 asyncio: mode=Mode.STRICT collected 16 items Fatal Python error: Segmentation fault Current thread 0x00007ffabc917b80 (most recent call first): File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py", line 113 in _call_store File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py", line 64 in init File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/c10d_rendezvous_backend.py", line 253 in create_backend File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/registry.py", line 36 in _create_c10d_handler File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/api.py", line 258 in create_handler File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/elastic/rendezvous/registry.py", line 66 in get_rendezvous_handler File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/launcher/api.py", line 238 in launch_agent File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/torch/distributed/launcher/api.py", line 135 in call File "/home/runner/work/flytekit/flytekit/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py", line 393 in _execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/exceptions/scopes.py", line [21](https://github.com/flyteorg/flytekit/actions/runs/8182775035/job/22374791820?pr=2237#step:7:22)9 in user_entry_point File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/exceptions/scopes.py", line 143 in f File "/home/runner/work/flytekit/flytekit/plugins/flytekit-kf-pytorch/flytekitplugins/kfpytorch/task.py", line 428 in execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/base_task.py", line 675 in dispatch_execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/base_task.py", line 388 in sandbox_execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/base_task.py", line 308 in local_execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/promise.py", line 1195 in flyte_entity_call_handler File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/base_task.py", line 331 in call File "/home/runner/work/flytekit/flytekit/plugins/flytekit-kf-pytorch/tests/test_elastic_task.py", line 57 in wf File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/exceptions/scopes.py", line 212 in user_entry_point File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/exceptions/scopes.py", line 143 in f File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/workflow.py", line 796 in execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/workflow.py", line 312 in local_execute File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/promise.py", line 1203 in flyte_entity_call_handler File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/flytekit/core/workflow.py", line 287 in call File "/home/runner/work/flytekit/flytekit/plugins/flytekit-kf-pytorch/tests/test_elastic_task.py", line 59 in test_end_to_end File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/python.py", line 194 in pytest_pyfunc_call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_callers.py", line 102 in _multicall File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_manager.py", line 119 in _hookexec File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_hooks.py", line 501 in call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/python.py", line 1831 in runtest File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 170 in pytest_runtest_call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_callers.py", line 102 in _multicall File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_manager.py", line 119 in _hookexec File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_hooks.py", line 501 in call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 263 in <lambda> File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 342 in from_call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 262 in call_runtest_hook File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line [22](https://github.com/flyteorg/flytekit/actions/runs/8182775035/job/22374791820?pr=2237#step:7:23)3 in call_and_report File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 134 in runtestprotocol File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/runner.py", line 115 in pytest_runtest_protocol File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_callers.py", line 102 in _multicall File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_manager.py", line 119 in _hookexec File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_hooks.py", line 501 in call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/main.py", line 352 in pytest_runtestloop File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_callers.py", line 102 in _multicall File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_manager.py", line 119 in _hookexec File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_hooks.py", line 501 in call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/main.py", line 327 in _main File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/main.py", line 273 in wrap_session File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/main.py", line 320 in pytest_cmdline_main File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_callers.py", line 102 in _multicall File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_manager.py", line 119 in _hookexec File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pluggy/_hooks.py", line 501 in call File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/config/__init__.py", line 175 in main File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/_pytest/config/__init__.py", line 198 in console_main File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/pytest/__main__.py", line 7 in <module> File "/opt/hostedtoolcache/Python/3.12.2/x64/lib/python3.12/site-packages/coverage/execfile.py", line 211 in run File "/opt/hostedtoolcache/P… flyteorg/flyte