fix another dbn config

erdogant · Oct 23, 2024 · d9432c4 · d9432c4
1 parent e5be8b2
commit d9432c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bnlearn/parameter_learning.py b/bnlearn/parameter_learning.py
@@ -115,7 +115,7 @@ def fit(model, df, methodtype='bayes', scoretype='bdeu', smooth=None, n_jobs=-1,
     independence_test = model.get('independence_test', None)
 
     # Automatically set methodtype to DBN
-    if str(model['config'].get('method')).lower()=='dbn':
+    if model.get('config', {}).get('method') == 'DBN':
         config['method'] = 'DBN'
         if verbose>=3: print('[bnlearn] >Methodtype is set to DynamicBayesianNetwork (DBN)')