python3Packages.holistic-trace-analysis: init at 0.5.0 (#421780)

{

  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 ];

  };

}

  holidays = callPackage ../development/python-modules/holidays { };

  holistic-trace-analysis = callPackage ../development/python-modules/holistic-trace-analysis { };

  hologram = callPackage ../development/python-modules/hologram { };

  holoviews = callPackage ../development/python-modules/holoviews { };