diff --git a/pkgs/development/python-modules/ultralytics-thop/default.nix b/pkgs/development/python-modules/ultralytics-thop/default.nix new file mode 100644 index 000000000000..224c4953ed69 --- /dev/null +++ b/pkgs/development/python-modules/ultralytics-thop/default.nix @@ -0,0 +1,47 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + + # build-system + setuptools, + + # dependencies + numpy, + torch, + + # tests + pytestCheckHook, +}: + +buildPythonPackage rec { + pname = "ultralytics-thop"; + version = "2.0.14"; + pyproject = true; + + src = fetchFromGitHub { + owner = "ultralytics"; + repo = "thop"; + tag = "v${version}"; + hash = "sha256-Xn01zh0/oaMPqH0FPDKElE0q7di3sDrXYcXCg6I/89E="; + }; + + build-system = [ setuptools ]; + + dependencies = [ + numpy + torch + ]; + + pythonImportsCheck = [ "thop" ]; + + nativeCheckInputs = [ pytestCheckHook ]; + + meta = { + homepage = "https://github.com/ultralytics/thop"; + changelog = "https://github.com/ultralytics/thop/releases/tag/v${version}"; + description = "Profile PyTorch models by computing the number of Multiply-Accumulate Operations (MACs) and parameters"; + license = lib.licenses.agpl3Only; + maintainers = with lib.maintainers; [ osbm ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 3da85187ea93..48be133d6db3 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -17934,6 +17934,8 @@ self: super: with self; { ultraheat-api = callPackage ../development/python-modules/ultraheat-api { }; + ultralytics-thop = callPackage ../development/python-modules/ultralytics-thop { }; + umalqurra = callPackage ../development/python-modules/umalqurra { }; umap-learn = callPackage ../development/python-modules/umap-learn { };