From b8393db284317016fe547fd713f7f5639af9e3f0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?AutoMath=C3=AFs?= <automathis@protonmail.com>
Date: Sun, 18 Aug 2024 11:39:56 +0200
Subject: [PATCH] python3Packages.hebg: init at 0.2.4

Co-authored-by: OTABI Tomoya <tomoya.otabi@gmail.com>
---
 .../python-modules/hebg/default.nix           | 56 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 58 insertions(+)
 create mode 100644 pkgs/development/python-modules/hebg/default.nix

diff --git a/pkgs/development/python-modules/hebg/default.nix b/pkgs/development/python-modules/hebg/default.nix
new file mode 100644
index 0000000000000..5bf209fe7a7ba
--- /dev/null
+++ b/pkgs/development/python-modules/hebg/default.nix
@@ -0,0 +1,56 @@
+{
+  lib,
+  stdenv,
+  buildPythonPackage,
+  fetchPypi,
+
+  # Runtime dependencies
+  networkx,
+  matplotlib,
+  numpy,
+  tqdm,
+  scipy,
+
+  # Build, dev and test dependencies
+  setuptools-scm,
+  pytestCheckHook,
+  pytest-check,
+  pytest-mock,
+}:
+
+buildPythonPackage rec {
+  pname = "hebg";
+  version = "0.2.4";
+  pyproject = true;
+
+  src = fetchPypi {
+    inherit pname version;
+    hash = "sha256-11bz+FbnaEVLiXT1eujMw8lvABlzVOeROOsdVgsyfjQ=";
+  };
+
+  build-system = [ setuptools-scm ];
+
+  dependencies = [
+    networkx
+    matplotlib
+    numpy
+    tqdm
+    scipy
+  ];
+
+  nativeCheckInputs = [
+    pytestCheckHook
+    pytest-check
+    pytest-mock
+  ];
+  env = lib.optionalAttrs stdenv.hostPlatform.isDarwin { MPLBACKEND = "Agg"; };
+  pythonImportsCheck = [ "hebg" ];
+
+  meta = {
+    description = "Hierachical Explainable Behaviors using Graphs";
+    homepage = "https://github.com/IRLL/HEB_graphs";
+    changelog = "https://github.com/IRLL/HEB_graphs/releases/tag/${version}";
+    license = lib.licenses.gpl3Only;
+    maintainers = with lib.maintainers; [ automathis ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 7b0200329f870..ec74c2ceb3063 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -5537,6 +5537,8 @@ self: super: with self; {
 
   heatzypy = callPackage ../development/python-modules/heatzypy { };
 
+  hebg = callPackage ../development/python-modules/hebg { };
+
   help2man = callPackage ../development/python-modules/help2man { };
 
   helpdev = callPackage ../development/python-modules/helpdev { };