Merge pull request #1252 from qiboteam/optional-deps

Introduce pip extras

.envrc

@@ -2,10 +2,8 @@ if ! has nix_direnv_version || ! nix_direnv_version 2.2.1; then
   source_url "https://raw.githubusercontent.com/nix-community/nix-direnv/2.2.1/direnvrc" "sha256-zelF0vLbEl5uaqrfIzbgNzJWGmLzCmYAkInj/LNxvKs="
 fi
 
-nix_direnv_watch_file devenv.nix
-nix_direnv_watch_file devenv.lock
-nix_direnv_watch_file devenv.yaml
-if ! use flake . --impure
-then
+watch_file flake.nix
+watch_file flake.lock
+if ! use flake . --impure; then
   echo "devenv could not be built. The devenv environment was not loaded. Make the necessary changes to devenv.nix and hit enter to try again." >&2
 fi

flake.nix

@@ -1,6 +1,6 @@
 {
   inputs = {
-    nixpkgs.url = "github:NixOS/nixpkgs/nixos-23.05";
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
     systems.url = "github:nix-systems/default";
     devenv.url = "github:cachix/devenv";
     nixpkgs-python.url = "github:cachix/nixpkgs-python";
@@ -35,34 +35,15 @@
             {
               packages = with pkgs; [
                 pre-commit
-                (python3Packages.buildPythonPackage
-                  (let
-                    pname = "poethepoet";
-                    version = "0.24.1";
-                  in {
-                    inherit pname version;
-                    src = fetchPypi {
-                      inherit pname version;
-                      sha256 = "sha256-OvpEtPxzJ98N2RLtoBJgSgcq8rtNJD+w5B6Oyo2r+e0=";
-                      python = "py3";
-                      dist = "py3";
-                      format = "wheel";
-                    };
-                    format = "wheel";
-                    propagatedBuildInputs = [
-                      # Specify dependencies
-                      pkgs.python3Packages.pastel
-                      pkgs.python3Packages.tomli
-                    ];
-                  }))
+                poethepoet
+                stdenv.cc.cc.lib
               ];
 
               languages.python = {
                 enable = true;
                 poetry = {
                   enable = true;
                   install.enable = true;
-                  install.allExtras = true;
                   install.groups = ["dev"];
                 };
                 version = "3.11";

pyproject.toml

@@ -28,7 +28,7 @@ cma = "^3.3.0"
 joblib = "^1.2.0"
 hyperopt = "^0.2.7"
 tabulate = "^0.9.0"
-openqasm3 = {version = ">=0.5.0", extras = ["parser"]}
+openqasm3 = { version = ">=0.5.0", extras = ["parser"] }
 numpy = "^1.26.4"
 networkx = "^3.2.1"
 cvxpy = { version = "^1.4.2", optional = true }
@@ -98,12 +98,15 @@ cuquantum-python-cu12 = "^23.3.0"
 qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
+[tool.poetry.extras]
+tensorflow = ["tensorflow"]
+qinfo = ["cvxpy"]
+
 [tool.pylint.reports]
 output-format = "colorized"
 
-
 [tool.coverage.run]
-omit = [ "src/qibo/noise_model.py" ]
+omit = ["src/qibo/noise_model.py"]
 
 [tool.pytest.ini_options]
 testpaths = ['tests/']