1 files changed, 47 insertions, 40 deletions
diff --git a/pkgs/development/python-modules/pgmpy/default.nix b/pkgs/development/python-modules/pgmpy/default.nix
index a953b575b591d..1217f573cdcd9 100644
--- a/pkgs/development/python-modules/pgmpy/default.nix
+++ b/pkgs/development/python-modules/pgmpy/default.nix
@@ -1,47 +1,43 @@
-{ lib
-, buildPythonPackage
-, fetchFromGitHub
-, pythonOlder
-# build inputs
-, networkx
-, numpy
-, scipy
-, scikit-learn
-, pandas
-, pyparsing
-, torch
-, statsmodels
-, tqdm
-, joblib
-, opt-einsum
-# check inputs
-, pytestCheckHook
-, pytest-cov
-, coverage
-, mock
-, black
+{
+  lib,
+  buildPythonPackage,
+  fetchFromGitHub,
+
+  # dependencies
+  networkx,
+  numpy,
+  scipy,
+  scikit-learn,
+  pandas,
+  pyparsing,
+  torch,
+  statsmodels,
+  tqdm,
+  joblib,
+  opt-einsum,
+  xgboost,
+  google-generativeai,
+
+  # tests
+  pytestCheckHook,
+  pytest-cov,
+  coverage,
+  mock,
+  black,
 }:
-let
+buildPythonPackage rec {
   pname = "pgmpy";
-  version = "0.1.25";
-  # optional-dependencies = {
-  #   all = [ daft ];
-  # };
-in
-buildPythonPackage {
-  inherit pname version;
-  format = "setuptools";
-
-  disabled = pythonOlder "3.7";
+  version = "0.1.26";
+  pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pgmpy";
-    repo = pname;
+    repo = "pgmpy";
     rev = "refs/tags/v${version}";
-    hash = "sha256-d2TNcJQ82XxTWdetLgtKXRpFulAEEzrr+cyRewoA6YI=";
+    hash = "sha256-RusVREhEXYaJuQXTaCQ7EJgbo4+wLB3wXXCAc3sBGtU=";
   };
 
-  propagatedBuildInputs = [
+  dependencies = [
     networkx
     numpy
     scipy
@@ -53,10 +49,21 @@ buildPythonPackage {
     tqdm
     joblib
     opt-einsum
+    xgboost
+    google-generativeai
   ];
 
   disabledTests = [
-    "test_to_daft" # requires optional dependency daft
+    # flaky:
+    # AssertionError: -45.78899127622197 != -45.788991276221964
+    "test_score"
+
+    # self.assertTrue(np.isclose(coef, dep_coefs[i], atol=1e-4))
+    # AssertionError: False is not true
+    "test_pillai"
+
+    # requires optional dependency daft
+    "test_to_daft"
   ];
 
   nativeCheckInputs = [
@@ -68,11 +75,11 @@ buildPythonPackage {
     black
   ];
 
-  meta = with lib; {
+  meta = {
     description = "Python Library for learning (Structure and Parameter), inference (Probabilistic and Causal), and simulations in Bayesian Networks";
     homepage = "https://github.com/pgmpy/pgmpy";
     changelog = "https://github.com/pgmpy/pgmpy/releases/tag/v${version}";
-    license = licenses.mit;
-    maintainers = with maintainers; [ happysalada ];
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ happysalada ];
   };
 }