Thanks but why does this behavior not happen locally? Also, the bundle config doesn't reference torch, it's used in code and included as a dev dependency in pyproject.toml. My libraries are just this:

          libraries:
            - whl: ../dist/*.whl
            - pypi:
                package: pydantic>=2.0
            - maven:
                coordinates: io.dataflint:spark_2.12:0.4.0