python312Packages.mmengine: 0.10.4 -> 0.10.5 (#343207)

This commit is contained in:
Peder Bergebakken Sundt 2024-09-26 17:42:55 +02:00 committed by GitHub
commit 6992be0889
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,24 +1,32 @@
{ {
lib, lib,
addict, stdenv,
buildPythonPackage, buildPythonPackage,
coverage,
fetchFromGitHub, fetchFromGitHub,
lmdb,
# build-system
setuptools,
# dependencies
addict,
matplotlib, matplotlib,
mlflow,
numpy, numpy,
opencv4, opencv4,
parameterized,
pytestCheckHook,
pythonOlder,
pyyaml, pyyaml,
rich, rich,
setuptools,
stdenv,
termcolor, termcolor,
torch,
yapf, yapf,
# checks
bitsandbytes,
coverage,
dvclive,
lion-pytorch,
lmdb,
mlflow,
parameterized,
pytestCheckHook,
transformers,
}: }:
buildPythonPackage rec { buildPythonPackage rec {
@ -26,13 +34,11 @@ buildPythonPackage rec {
version = "0.10.5"; version = "0.10.5";
pyproject = true; pyproject = true;
disabled = pythonOlder "3.7";
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "open-mmlab"; owner = "open-mmlab";
repo = "mmengine"; repo = "mmengine";
rev = "refs/tags/v${version}"; rev = "refs/tags/v${version}";
hash = "sha256-+YDtYHp3BwKvzhmHC6hAZ3Qtc9uRZMo/TpWqdpm2hn0="; hash = "sha256-bZ6O4UOYUCwq11YmgRWepOIngYxYD/fNfM/VmcyUv9k=";
}; };
build-system = [ setuptools ]; build-system = [ setuptools ];
@ -49,12 +55,15 @@ buildPythonPackage rec {
]; ];
nativeCheckInputs = [ nativeCheckInputs = [
# bitsandbytes (broken as of 2024-07-06)
coverage coverage
dvclive
lion-pytorch
lmdb lmdb
mlflow mlflow
parameterized parameterized
pytestCheckHook pytestCheckHook
torch transformers
]; ];
preCheck = preCheck =
@ -108,12 +117,12 @@ buildPythonPackage rec {
"test_close" "test_close"
]; ];
meta = with lib; { meta = {
description = "Library for training deep learning models based on PyTorch"; description = "Library for training deep learning models based on PyTorch";
homepage = "https://github.com/open-mmlab/mmengine"; homepage = "https://github.com/open-mmlab/mmengine";
changelog = "https://github.com/open-mmlab/mmengine/releases/tag/v${version}"; changelog = "https://github.com/open-mmlab/mmengine/releases/tag/v${version}";
license = with licenses; [ asl20 ]; license = with lib.licenses; [ asl20 ];
maintainers = with maintainers; [ rxiao ]; maintainers = with lib.maintainers; [ rxiao ];
broken = broken =
stdenv.hostPlatform.isDarwin || (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64); stdenv.hostPlatform.isDarwin || (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64);
}; };