Unverified Commit d042fb41 authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.sparsezoo: drop (#439157)

parents 9ac9bb01 c82b20e9
Loading
Loading
Loading
Loading
+0 −121
Original line number Diff line number Diff line
{
  lib,
  buildPythonPackage,
  fetchFromGitHub,

  # build-system
  setuptools,

  # dependencies
  click,
  geocoder,
  numpy,
  onnx,
  pandas,
  protobuf,
  py-machineid,
  pydantic,
  pyyaml,
  requests,
  tqdm,

  # checks
  matplotlib,
  pytestCheckHook,
}:

buildPythonPackage rec {
  pname = "sparsezoo";
  version = "1.9.0";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "neuralmagic";
    repo = "sparsezoo";
    tag = "v${version}";
    hash = "sha256-eMP/whm06QX5x/RBoYsYwuKxFnpFmqlgh2uDsI3Vaog=";
  };

  build-system = [ setuptools ];

  pythonRelaxDeps = [
    "numpy"
    "onnx"
  ];

  dependencies = [
    click
    geocoder
    numpy
    onnx
    pandas
    protobuf
    py-machineid
    pydantic
    pyyaml
    requests
    tqdm
  ];

  pythonImportsCheck = [ "sparsezoo" ];

  nativeCheckInputs = [
    matplotlib
    pytestCheckHook
  ];

  disabledTests = [
    # Require network access
    "test_analysis"
    "test_custom_default"
    "test_draw_operation_chart"
    "test_draw_parameter_chart"
    "test_draw_parameter_operation_combined_chart"
    "test_draw_sparsity_by_layer_chart"
    "test_extract_node_id"
    "test_fail_default_on_empty"
    "test_folder_structure"
    "test_from_model_analysis"
    "test_generate_outputs"
    "test_get_layer_and_op_counts"
    "test_get_node_bias"
    "test_get_node_bias_name"
    "test_get_node_four_block_sparsity"
    "test_get_node_num_four_block_zeros_and_size"
    "test_get_node_num_zeros_and_size"
    "test_get_node_sparsity"
    "test_get_node_weight_and_shape"
    "test_get_node_weight_name"
    "test_get_ops_dict"
    "test_get_zero_point"
    "test_graphql_api_response"
    "test_is_four_block_sparse_layer"
    "test_is_parameterized_prunable_layer"
    "test_is_quantized_layer"
    "test_is_sparse_layer"
    "test_latency_extractor"
    "test_load_files_from_stub"
    "test_model_from_stub"
    "test_model_gz_extraction_from_local_files"
    "test_model_gz_extraction_from_stub"
    "test_recipe_getters"
    "test_search_models"
    "test_setup_model_from_objects"
    "test_setup_model_from_paths"
    "test_validate"
  ];

  disabledTestPaths = [
    # Require network access
    "tests/sparsezoo/analyze/test_model_analysis_creation.py"
    "tests/sparsezoo/deployment_package/utils/test_utils.py"
  ];

  meta = {
    description = "Neural network model repository for highly sparse and sparse-quantized models with matching sparsification recipes";
    homepage = "https://github.com/neuralmagic/sparsezoo";
    changelog = "https://github.com/neuralmagic/sparsezoo/releases/tag/${src.tag}";
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ GaetanLepage ];
  };
}
+1 −0
Original line number Diff line number Diff line
@@ -768,6 +768,7 @@ mapAliases ({
  spacy_models = spacy-models; # added 2024-04-21
  SPARQLWrapper = sparqlwrapper;
  spark_parser = spark-parser; # added 2024-01-07
  sparsezoo = throw "sparsezoo has been removed since it is abandonned"; # added 2025-09-01
  sphinx-jquery = sphinxcontrib-jquery; # added 2023-02-24
  sphinx_pypi_upload = throw "sphinx_pypi_upload has been removed since it is abandoned."; # added 2023-10-11
  sphinx_rtd_theme = sphinx-rtd-theme; # added 2022-08-03
+0 −2
Original line number Diff line number Diff line
@@ -17062,8 +17062,6 @@ self: super: with self; {
  sparse = callPackage ../development/python-modules/sparse { };
  sparsezoo = callPackage ../development/python-modules/sparsezoo { };
  spatial-image = callPackage ../development/python-modules/spatial-image { };
  spatialmath-python = callPackage ../development/python-modules/spatialmath-python { };