Fixes minor issues

pgmpy · Mar 14, 2023 · 957232a · 957232a
1 parent c0fd34a
commit 957232a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pgmpy/models/DynamicBayesianNetwork.py b/pgmpy/models/DynamicBayesianNetwork.py
@@ -905,7 +905,7 @@ def fit(self, data, estimator="MLE", state_names={}, n_jobs=-1):
 
             # Fit or fit_update with df_slice depending on the time slice
             if t_slice == 0:
-                const_bn.fit(df_slice, state_names={}, n_jobs=n_jobs)
+                const_bn.fit(df_slice, state_names=state_names, n_jobs=n_jobs)
             else:
                 const_bn.fit_update(
                     df_slice, n_prev_samples=t_slice * n_samples, n_jobs=n_jobs