Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.0 kB · 49 lines
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950{ buildPythonPackage, fetchFromGitHub, jupyterlab, lib, numpy, pandas, plotly, pydot, pytestCheckHook, setuptools, torch,}:
buildPythonPackage rec { pname = "HolisticTraceAnalysis"; version = "0.5.0"; pyproject = true;
src = fetchFromGitHub { owner = "facebookresearch"; repo = "HolisticTraceAnalysis"; tag = "v${version}"; hash = "sha256-3DuoP9gQ0vLlAAJ2uWw/oOEH/DTbn2xulzvqk4W3BiY="; };
build-system = [ setuptools ];
dependencies = [ jupyterlab numpy pandas plotly pydot torch ];
nativeCheckInputs = [ pytestCheckHook ];
pythonImportsCheck = [ "hta" ];
meta = { description = "Performance analysis tool to identify bottlenecks in distributed training workloads"; homepage = "https://github.com/facebookresearch/HolisticTraceAnalysis"; changelog = "https://github.com/facebookresearch/HolisticTraceAnalysis/releases/tag/v${version}"; license = lib.licenses.mit; maintainers = with lib.maintainers; [ jherland ]; };}