CI GPU

.github/workflows/python.yaml

@@ -133,20 +133,27 @@ jobs:
  - name: Install on GPU
  if: ${{ matrix.module == 'gpu' }}
  run: |
- source activate xoscar-test
+ source activate ${{ env.CONDA_ENV }}
+ which python
+ conda install -y conda-forge::nccl
  pip install --extra-index-url=https://pypi.nvidia.com cudf-cu12
  pip install ucx-py-cu12 cloudpickle psutil tblib uvloop packaging "numpy<2.0.0" scipy cython coverage flaky
- python setup.py clean --all
- pip install -e ./
+ # python setup.py clean --all
+ # pip install -e ./
  working-directory: ./python
 
  - name: Test with pytest
- if: ${{ matrix.module != 'gpu' }}
+ env:
+ MODULE: ${{ matrix.module }}
  run: |
- pytest --timeout=1500 \
- -W ignore::PendingDeprecationWarning \
- --cov-config=setup.cfg --cov-report=xml \
- --cov=xoscar xoscar --capture=no
+ if [[ "$MODULE" == "xoscar" ]]; then
+ pytest --timeout=1500 \
+ -W ignore::PendingDeprecationWarning \
+ --cov-config=setup.cfg --cov-report=xml --cov=xoscar xoscar --capture=no
+ else
+ source activate ${{ env.CONDA_ENV }}
+ pytest -m cuda --cov-config=setup.cfg --cov-report=xml --cov=xoscar --capture=no
+ fi
  working-directory: ./python
 
  - name: Test with pytest GPU