Disko

Add input to flake.nix:

disko.url = "github:nix-community/disko";
disko.inputs.nixpkgs.follows = "nixpkgs";

flake.nix:inputs

Add the module to modules list of nixosSystem calls:

inputs.disko.nixosModules.disko

nixos