Issue trying to install pytorch geometric

[j-bac]

I can't figure out how to specify these packages in pixi.toml:

torch = "*"
torch_geometric = "*"
torch_scatter = "*"
torch_sparse = "*" 

Install fails with:

Error:   × failed to solve the pypi requirements of environment 'gnn' for platform 'linux-64'
  ├─▶ failed to resolve pypi dependencies
  ├─▶ Failed to download and build `torch-scatter==2.1.2`
  ├─▶ The build backend returned an error
  ╰─▶ Call to `setuptools.build_meta:__legacy__.build_wheel` failed (exit status: 1)
      
      [stderr]
      Traceback (most recent call last):
        File "<string>", line 14, in <module>
        File "/work/PRTNR/CHUV/DIR/rgottar1/spatial/env/jbac/.cache/rattler/cache/uv-cache/builds-v0/.tmpyEj3qi/lib/python3.11/site-packages/setuptools/build_meta.py", line 333, in
      get_requires_for_build_wheel
          return self._get_build_requires(config_settings, requirements=[])
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
        File "/work/PRTNR/CHUV/DIR/rgottar1/spatial/env/jbac/.cache/rattler/cache/uv-cache/builds-v0/.tmpyEj3qi/lib/python3.11/site-packages/setuptools/build_meta.py", line 301, in _get_build_requires
          self.run_setup()
        File "/work/PRTNR/CHUV/DIR/rgottar1/spatial/env/jbac/.cache/rattler/cache/uv-cache/builds-v0/.tmpyEj3qi/lib/python3.11/site-packages/setuptools/build_meta.py", line 518, in run_setup
          super().run_setup(setup_script=setup_script)
        File "/work/PRTNR/CHUV/DIR/rgottar1/spatial/env/jbac/.cache/rattler/cache/uv-cache/builds-v0/.tmpyEj3qi/lib/python3.11/site-packages/setuptools/build_meta.py", line 317, in run_setup
          exec(code, locals())
        File "<string>", line 8, in <module>
      ModuleNotFoundError: No module named 'torch'
      
      hint: This error likely indicates that `torch-scatter@2.1.2` depends on `torch`, but doesn't declare it as a build dependency. If `torch-scatter` is a first-party package, consider adding `torch`
      to its `build-system.requires`. Otherwise, either add it to your `pyproject.toml` under:
      
      [tool.uv.extra-build-dependencies]
      torch-scatter = ["torch"]
      
      or `uv pip install torch` into the environment and re-run with `--no-build-isolation`.

The original install documentation requires using -f options but I am also not sure how to specify this
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html

I tried adding this to pixi.toml but it didn't solve the problem:

[pypi-options]
find-links = [{url= 'https://data.pyg.org/whl/torch-2.8.0+cu129.html'}]