master-thesis/python/energy_flow_proper/05_gaussian_two_baths/flake.nix

{
  description = "Energy flow with a two HOs coupled to two baths and each other.";

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

  outputs = { self, utils, nixpkgs, ... }:
    (utils.lib.poetry2nixWrapper nixpkgs {
      name = "05_gaussian_twobath";
      shellPackages = pkgs: with pkgs; [ pyright python39Packages.jupyter python39Packages.jupyter ];
      python = pkgs: pkgs.python39Full;
      shellOverride = (oldAttrs: {
        shellHook = ''
                    export PYTHONPATH=/home/hiro/src/hopsflow/:$PYTHONPATH
                    '';
      });

      extraOverrides = self: super: {
        numba = super.numba.override (
          {
            preferWheel = false;
          });
      };

      noPackage = true;
      poetryArgs = {
        projectDir = ./.;
      };
    });
}