bugfix issue105 DBN error

erdogant · Oct 23, 2024 · 896052f · 896052f
1 parent f89d8eb
commit 896052f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/bnlearn/bnlearn.py b/bnlearn/bnlearn.py
@@ -1139,7 +1139,7 @@ def plot_graphviz(model,
     if model['adjmat'].sum().sum()==0:
         if verbose>=3: print('[bnlearn]> Nothing to plot because no edges are present between nodes. ')
         return None
-    if model['config']['method']=='DBN':
+    if model.get('config', {}).get('method') == 'DBN':
         if verbose>=3: print('[bnlearn]> DynamicBayesianNetwork (DBN) can not be plot with Graphviz.')
         return None
 
@@ -1286,7 +1286,7 @@ def plot(model,
         if verbose>=3: print('[bnlearn]> Nothing to plot because no edges are present between nodes. ')
         return None
 
-    if model['config']['method']=='DBN' and interactive:
+    if model.get('config', {}).get('method') == 'DBN' and interactive:
         if verbose>=3: print('[bnlearn]> DynamicBayesianNetwork (DBN) can not be plot with Graphviz.')
         return None
 

diff --git a/bnlearn/examples.py b/bnlearn/examples.py
@@ -1,3 +1,11 @@
+#%%
+import bnlearn as bn
+model = bn.import_DAG('asia')
+G = bn.plot(model)
+
+
+
+
 # %% Impute categorical values
 import bnlearn as bn
 import pandas as pd