Bug: serialization issues after torch 2.6

[fabiocat93]

Description

on jan 29, torch 2.6 introduced some disruptive changes for users serializing more than basic types (https://github.com/pytorch/pytorch/releases/tag/v2.6.0) and, consequently, our pydra tests src/tests/utils/data_structures/pydra_helpers_test.py fail. as a workaround, for now, I have fixed torch version to < 2.6. @wilke0818 and I have been exploring the space but haven't found a solution yet.

Steps to Reproduce

Install torch 2.6 and run unit tests

Expected Results

Success

Actual Results

================================================== FAILURES ===================================================
_________________________________________________ test_pydra __________________________________________________
[gw2] darwin -- Python 3.10.10 /Users/fabiocat/Library/Caches/pypoetry/virtualenvs/senselab-KP8v1V64-py3.10/bin/python

    def test_pydra() -> None:
        """Test simple tensor pydra workflow."""
        wf = pydra.Workflow(name="wf_test", input_spec=["x"])
        wf.split("x", x=[torch.tensor([[3, 4], [5, 6]]), torch.tensor([[0, 1], [1, 2]])])
    
        wf.add(pydra_task(name="test_task_task", test_input=wf.lzin.x))
>       wf.set_output([("wf_out", wf.test_task_task.lzout.out)])

src/tests/utils/data_structures/pydra_helpers_test.py:19: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

self = LazyOut(_task=<pydra.engine.task.FunctionTask object at 0x3730e6500>), name = 'out'

    def __getattr__(self, name):
        if name in ("_task", "_attr_type", "_field_names"):
            raise AttributeError(f"{name} hasn't been set yet")
        if name not in self._field_names:
>           raise AttributeError(
                f"Task '{self._task.name}' has no {self._attr_type} attribute '{name}', "
                "available: '" + "', '".join(self._field_names) + "'"
            )
E           AttributeError: Task 'test_task_task' has no output attribute 'out', available: '_is_param', 'all_'

../../Library/Caches/pypoetry/virtualenvs/senselab-KP8v1V64-py3.10/lib/python3.10/site-packages/pydra/engine/specs.py:696: AttributeError

Additional Notes

No response