mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 23:12:07 +00:00
9ed0c85fe1
To make sure that everything is formatted with the same black version as CI. I sometimes use isort for new files to get nicely ordered imports, so add it as well. Also set the isort configuration to format in a way that is compatible with black.
59 lines
916 B
Nix
59 lines
916 B
Nix
{
|
|
mkShell,
|
|
black,
|
|
isort,
|
|
openssl,
|
|
pkg-config,
|
|
protobuf,
|
|
python3,
|
|
pyright,
|
|
redocly,
|
|
ruff,
|
|
rust-bin,
|
|
server,
|
|
}:
|
|
|
|
mkShell {
|
|
buildInputs =
|
|
[
|
|
black
|
|
isort
|
|
openssl.dev
|
|
pkg-config
|
|
(rust-bin.stable.latest.default.override {
|
|
extensions = [
|
|
"rust-analyzer"
|
|
"rust-src"
|
|
];
|
|
})
|
|
protobuf
|
|
pyright
|
|
redocly
|
|
ruff
|
|
]
|
|
++ (with python3.pkgs; [
|
|
venvShellHook
|
|
docker
|
|
pip
|
|
ipdb
|
|
click
|
|
pytest
|
|
pytest-asyncio
|
|
syrupy
|
|
]);
|
|
|
|
inputsFrom = [ server ];
|
|
|
|
venvDir = "./.venv";
|
|
|
|
postVenvCreation = ''
|
|
unset SOURCE_DATE_EPOCH
|
|
( cd server ; python -m pip install --no-dependencies -e . )
|
|
( cd clients/python ; python -m pip install --no-dependencies -e . )
|
|
'';
|
|
postShellHook = ''
|
|
unset SOURCE_DATE_EPOCH
|
|
export PATH=$PATH:~/.cargo/bin
|
|
'';
|
|
}
|