master-thesis/python/energy_flow_proper/08_dynamic_one_bath/flake.nix

{
  description = "Testing out the one bath case with time dependent coupling.";

  inputs = {
    nixpkgs.url = "nixpkgs/nixos-unstable";
    utils.url = "github:vale981/hiro-flake-utils";
  };

  outputs = { self, utils, nixpkgs, ... }:
    (utils.lib.poetry2nixWrapper nixpkgs {
      name = "08_dynamic_one_bath";
      extraOverlay = (self: super: {
        flint = super.flint.overrideAttrs (old: rec {
          pname = "flint";
          version = "39ac8817a18710effd96dff38770f4d1c80ad608";
          doCheck = false;
          src = super.pkgs.fetchFromGitHub {
            owner = "wbhart";
            repo = "flint2";
            rev = version;
            sha256 = "sha256-UIdKUT2oSNth6AHHRPZOTLR7gHnxgmxKWdOR4klPPLw=";
          };
        });

        arb = super.arb.overrideAttrs (old: rec {
          pname = "arb-git";
          version = "4e879e5602a59c14780085900236c9fa141f993d";
          doCheck = false;
          src = super.pkgs.fetchFromGitHub {
            owner = "fredrik-johansson";
            repo = "arb";
            rev = version;
            sha256 = "sha256-JoBZZk824+O5kVctdUv0qzlX9l3X2RWr5/NPedx50oQ=";
          };
        });
      });
      shellPackages = (pkgs:
      (with pkgs;
      [ pyright python39Packages.jupyter sshfs arb]));

      python = pkgs: pkgs.python39Full;
      shellOverride = (pkgs: oldAttrs: {
        shellHook = ''
                    export PYTHONPATH=/home/hiro/src/two_qubit_model/:$PYTHONPATH
                    export PYTHONPATH=/home/hiro/src/hops/:$PYTHONPATH
                    export PYTHONPATH=/home/hiro/src/hopsflow/:$PYTHONPATH
                    export PYTHONPATH=/home/hiro/src/stocproc/:$PYTHONPATH

                    export LD_LIBRARY_PATH="${(pkgs.lib.makeLibraryPath [pkgs.arb])}"
      #               '';
      });
      noPackage = true;
      poetryArgs = {
        projectDir = ./.;
      };
    });
}