testing workflows with GPU

Author: lorenzotomada

.github/workflows/docs.yml

@@ -6,6 +6,7 @@ on:
       - main
       - final_project
       - gpu_branch
+      - workflows
 
   pull_request:
     branches:

.github/workflows/tests.yml

@@ -6,6 +6,7 @@ on:
       - main
       - final_project
       - gpu_branch
+      - workflows
 
   pull_request:
     branches:
@@ -18,24 +19,30 @@ jobs:
     name: Testing using pytest
     runs-on: ubuntu-latest
     container:
-      image: cupy/cupy:v13.4.0
+      image: nvidia/cuda:12.1.0-devel-ubuntu22.04
 
     steps:
     - uses: actions/checkout@v3
 
-    - name: Install system dependencies
+    - name: Set up Python
+      uses: actions/setup-python@v3
+      with:
+        python-version: 3.10.16
+    
+    - name: Set up Python environment
       run: |
-        apt-get update && apt-get install -y python3-venv python3-pip
+        python -m venv venv
+        source venv/bin/activate
+      shell: bash -e {0}
 
-    - name: Install additional dependencies
+    - name: Install dependencies
       run: |
-        python3 -m venv venv
         source venv/bin/activate
         python -m pip install .
       shell: bash -e {0}
 
     - name: Test with pytest
       run: |
         source venv/bin/activate
-        python -m pytest
+        python -m pytest -v
       shell: bash -e {0}

src/pyclassify/eigenvalues.py

@@ -81,10 +81,8 @@ def eigenvalues_cp(A):
     This function checks if the input matrix is square and symmetric, then computes its eigenvalues using
     CuPy's sparse linear algebra solvers. It uses `eigsh` for more efficient computation.
     IMPORTANT: it important to underline a couple of things regarding this function:
-    - installing using the command
-    .. code-block:: shell
-            python -m pip install cupy-cuda12x
-      does not allow, for some reason, to import cupyx.scipy.sparse.linalg, and it necessary to import the
+    - installing using the command 'python -m pip install cupy-cuda12x' does not allow, for some reason,
+      to import cupyx.scipy.sparse.linalg, and it necessary to import the
       function manually form the source code. The problem can be observed using python3.10, while for
       python3.12 things seem to work.
     - the eighs function in this case does not allow to compute *all* the eigenvalues, but only a number