feat: fixed pyproject and publish workflow

Author: BrunoLiegiBastonLiegi

.github/workflows/publish.yml

@@ -3,7 +3,9 @@ name: docs
 on:
   workflow_dispatch:
   push:
-
+    branches: [main]
+    tags:
+      - "*"
 jobs:
   evaluate-label:
     runs-on: ubuntu-latest
@@ -21,7 +23,7 @@ jobs:
               echo "No deployment for this ref" >&2
               exit 1
           fi
-          
+
 
   deploy-docs:
     needs: [evaluate-label]

pyproject.toml

@@ -14,8 +14,8 @@ documentation = "https://qibo.science/docs/qiboml/stable"
 packages = [{ include = "qiboml", from = "src" }]
 
 [tool.poetry.dependencies]
-python = ">=3.9,<3.13"
-numpy = "^1.26.4"
+python = ">=3.10,<3.13"
+numpy = "^2.0.0"
 keras = { version = "^3.0.0", optional = true }
 tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'", optional = true }
 # TODO: the marker is a temporary solution due to the lack of the tensorflow-io 0.32.0's wheels for Windows, this package is one of
@@ -62,6 +62,19 @@ torch = { version = "^2.3.1"}
 [tool.poetry.group.benchmark.dependencies]
 pytest-benchmark = { version = "^4.0.0", extras = ["histogram"] }
 
+[tool.poetry.group.cuda11]
+optional = true
+
+[tool.poetry.group.cuda11.dependencies]
+jax = {extras = ["cuda11"], version = "^0.4.25"}
+
+[tool.poetry.group.cuda12]
+optional = true
+
+[tool.poetry.group.cuda12.dependencies]
+jax = {extras = ["cuda12"], version = "^0.4.25"}
+
+
 [tool.poe.tasks]
 bench = "pytest benchmarks/"
 test = "pytest"