python313Packages.pgmpy: adjust check section (#446783)

pkgs/development/python-modules/pgmpy/default.nix

@@ -4,19 +4,20 @@
   fetchFromGitHub,
 
   # dependencies
+  google-generativeai,
+  joblib,
   networkx,
   numpy,
-  scipy,
-  scikit-learn,
+  opt-einsum,
   pandas,
   pyparsing,
-  torch,
+  pyro-ppl,
+  scikit-learn,
+  scipy,
   statsmodels,
+  torch,
   tqdm,
-  joblib,
-  opt-einsum,
   xgboost,
-  google-generativeai,
 
   # tests
   pytestCheckHook,
@@ -38,19 +39,20 @@ buildPythonPackage rec {
   };
 
   dependencies = [
+    google-generativeai
+    joblib
     networkx
     numpy
-    scipy
-    scikit-learn
+    opt-einsum
     pandas
     pyparsing
-    torch
+    pyro-ppl
+    scikit-learn
+    scipy
     statsmodels
+    torch
     tqdm
-    joblib
-    opt-einsum
     xgboost
-    google-generativeai
   ];
 
   disabledTests = [
@@ -64,6 +66,9 @@ buildPythonPackage rec {
 
     # requires optional dependency daft
     "test_to_daft"
+
+    # AssertionError
+    "test_estimate_example_smoke_test"
   ];
 
   nativeCheckInputs = [
@@ -75,6 +80,8 @@ buildPythonPackage rec {
     black
   ];
 
+  pythonImportsCheck = [ "pgmpy" ];
+
   meta = {
     description = "Python Library for learning (Structure and Parameter), inference (Probabilistic and Causal), and simulations in Bayesian Networks";
     homepage = "https://github.com/pgmpy/pgmpy";