python312Packages.torchprofile: init at 0.0.3

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  setuptools,

  numpy,

  torch,

  torchvision,

}:

buildPythonPackage rec {

  pname = "torchprofile";

  version = "0.0.3";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "zhijian-liu";

    repo = "torchprofile";

    rev = "refs/tags/v${version}";

    hash = "sha256-6vxZHQwBjKpy288wcANdJ9gmvIOZloLv+iN76TtqYAI=";

  };

  build-system = [

    setuptools

  ];

  pythonRelaxDeps = [

    "torchvision"

  ];

  dependencies = [

    numpy

    torch

    torchvision

  ];

  pythonImportsCheck = [

    "torchprofile"

  ];

  meta = {

    changelog = "https://github.com/zhijian-liu/torchprofile/releases/tag/v${version}";

    description = "General and accurate MACs / FLOPs profiler for PyTorch models";

    homepage = "https://github.com/zhijian-liu/torchprofile";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ drupol ];

  };

}

  # Used by streamlit, 2021-01-29

  tornado_5 = callPackage ../development/python-modules/tornado/5.nix { };

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

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

  torrent-parser = callPackage ../development/python-modules/torrent-parser { };