expman 0.5.1

High-performance experiment manager with Rust backend
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153

{
  description = "expman-rs: High-performance experiment manager in Rust";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    fenix = {
      url = "github:nix-community/fenix";
      inputs.nixpkgs.follows = "nixpkgs";
    };
    flake-utils.url = "github:numtide/flake-utils";
  };

  nixConfig = {
    extra-substituters = [ "https://lokeshmohanty.cachix.org" ];
    extra-trusted-public-keys = [
      "lokeshmohanty.cachix.org-1:XkCPbX2XsKzlr0P/MecvqruyTeOA8SzJzwMcCOfuLuI="
    ];
  };

  outputs =
    {
      self,
      nixpkgs,
      fenix,
      flake-utils,
      ...
    }:
    flake-utils.lib.eachDefaultSystem (
      system:
      let
        pkgs = import nixpkgs {
          inherit system;
          overlays = [ fenix.overlays.default ];
        };

        # Rust toolchain: stable with extras
        rustToolchain = pkgs.fenix.stable.withComponents [
          "cargo"
          "clippy"
          "rust-src"
          "rustc"
          "rustfmt"
          "rust-analyzer"
        ];

        # Combined toolchain with WASM target
        fullToolchain = pkgs.fenix.combine [
          rustToolchain
          pkgs.fenix.targets.wasm32-unknown-unknown.stable.rust-std
        ];

        # Base Python for uv
        pythonBase = pkgs.python312;

      in
      {
        devShells.default = pkgs.mkShell {
          name = "expman-rs";

          packages = [
            fullToolchain
            pkgs.pkg-config
            pkgs.openssl
            pkgs.just
            pkgs.trunk
            pkgs.wasm-bindgen-cli
            pkgs.uv
            pkgs.maturin
            pkgs.protobuf
            pkgs.cargo-nextest
          ];

          RUST_LOG = "debug";
          RUST_BACKTRACE = "1";
          PYO3_PYTHON = "${pythonBase}/bin/python3";

          shellHook = ''
            echo "🦀 expman-rs dev environment"
          '';
        };

        packages = rec {
          # Rust CLI package (Backend + Integrated Frontend build)
          expman = pkgs.rustPlatform.buildRustPackage {
            pname = "expman";
            version = "0.5.1";
            src = ./.;
            cargoLock.lockFile = ./Cargo.lock;

            buildInputs = [ pkgs.openssl ];
            nativeBuildInputs = [
              pkgs.pkg-config
              pkgs.lld
              pkgs.trunk
              pkgs.wasm-bindgen-cli
              pkgs.binaryen
              pkgs.protobuf
              fullToolchain
            ];

            # Build the frontend before the main package
            # Setting TRUNK_OFFLINE=true ensures it uses the Nix-provided wasm-bindgen-cli
            preBuild = ''
              export HOME=$TMPDIR
              export TRUNK_OFFLINE=true
              export TRUNK_BUILD_WASM_OPT=false
              trunk build --release
            '';

            buildFeatures = [
              "cli"
              "server"
            ];
            cargoBuildFlags = [
              "--bin"
              "exp"
            ];

            # Tests are handled in CI
            doCheck = false;
          };

          # Python package
          python3Packages = {
            expman-rs = pkgs.python3.pkgs.buildPythonPackage {
              pname = "expman-rs";
              version = "0.5.1";
              format = "pyproject";
              src = ./.;
              postPatch = "cp ../../Cargo.lock .";
              postUnpack = ''
                export sourceRoot=$sourceRoot/wrappers/python
              '';
              nativeBuildInputs = [
                pkgs.maturin
                pkgs.rustPlatform.maturinBuildHook
                pkgs.rustPlatform.cargoSetupHook
                pkgs.cargo
                pkgs.rustc
              ];
              buildInputs = [ pkgs.openssl ];
              cargoDeps = pkgs.rustPlatform.importCargoLock {
                lockFile = ./Cargo.lock;
              };
            };
          };

          exp = expman;
          default = exp;
        };
      }
    );
}