diff --git a/pkgs/development/python-modules/pyluwen/default.nix b/pkgs/development/python-modules/pyluwen/default.nix index 329e5ecb898a..dad420df0e76 100644 --- a/pkgs/development/python-modules/pyluwen/default.nix +++ b/pkgs/development/python-modules/pyluwen/default.nix @@ -34,6 +34,7 @@ buildPythonPackage rec { postPatch = '' cd ../$sourceRoot cp --no-preserve=ownership,mode ../../Cargo.lock . + sed -i '0,/version = /{s/version = "*.*.*"/version = "${version}"/g}' Cargo.toml ''; nativeBuildInputs = with rustPlatform; [ 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/development/python-modules/tt-tools-common/default.nix b/pkgs/development/python-modules/tt-tools-common/default.nix index a6e69f9bc319..bdff5a7aa6ee 100644 --- a/pkgs/development/python-modules/tt-tools-common/default.nix +++ b/pkgs/development/python-modules/tt-tools-common/default.nix @@ -13,21 +13,23 @@ requests, tqdm, pydantic, + setuptools-scm, }: buildPythonPackage rec { pname = "tt-tools-common"; - version = "1.4.25"; + version = "1.4.28"; pyproject = true; src = fetchFromGitHub { owner = "tenstorrent"; repo = "tt-tools-common"; tag = "v${version}"; - hash = "sha256-phal8KxfQqsGAIcKQTlSPZB04J158jZYlyamZr45vdU="; + hash = "sha256-L7MYrzQXb0LakQCx5CMlgTo0EjBoLLKC8u38eXksvoo="; }; build-system = [ setuptools + setuptools-scm ]; dependencies = [ diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 4703021f89a8..3ffc11747bd0 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -18833,6 +18833,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 { };