diff --git a/pkgs/development/python-modules/tt-smi/default.nix b/pkgs/development/python-modules/tt-smi/default.nix new file mode 100644 index 000000000000..c58ae346e466 --- /dev/null +++ b/pkgs/development/python-modules/tt-smi/default.nix @@ -0,0 +1,59 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + pythonOlder, + setuptools, + distro, + elasticsearch, + pydantic, + pyluwen, + rich, + textual, + pre-commit, + importlib-resources, + tt-tools-common, + tomli, +}: +buildPythonPackage rec { + pname = "tt-smi"; + version = "3.0.30"; + pyproject = true; + + src = fetchFromGitHub { + owner = "tenstorrent"; + repo = "tt-smi"; + tag = "v${version}"; + hash = "sha256-C6CfcS0H3rFew/Y1uhmzICdFp1UYU7H9h3YPeAKlcbE="; + }; + + disabled = pythonOlder "3.13"; + + build-system = [ + setuptools + ]; + + dependencies = [ + distro + elasticsearch + pydantic + pyluwen + rich + textual + pre-commit + importlib-resources + tt-tools-common + setuptools + tomli + ]; + + # Fails due to having no tests + dontUsePytestCheck = true; + + meta = { + description = "Tenstorrent console based hardware information program"; + homepage = "https://github.com/tenstorrent/tt-smi"; + maintainers = with lib.maintainers; [ RossComputerGuy ]; + license = with lib.licenses; [ asl20 ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index f2747156e712..2534be2f00cd 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -18768,6 +18768,8 @@ self: super: with self; { tt-flash = callPackage ../development/python-modules/tt-flash { }; + tt-smi = callPackage ../development/python-modules/tt-smi { }; + tt-tools-common = callPackage ../development/python-modules/tt-tools-common { }; ttach = callPackage ../development/python-modules/ttach { };