Upload bayesian_network_interface.py

bayesian_network_interface.py

@@ -79,31 +79,32 @@ class AutonomyBayesianNetwork:
             'categories': {}
         }
 
-        if self.pyagrum_model:
-            for node in self.pyagrum_model.names():
+        # Prioriser pgmpy_model car il est plus fiable
+        if self.pgmpy_model:
+            for node in self.pgmpy_model.nodes():
                 category = self._get_variable_category(node)
                 structure['nodes'].append({
                     'name': node,
-                    'category': category,
-                    'domain': list(self.pyagrum_model.variable(node).labels())
+                    'category': category
                 })
                 structure['categories'][node] = category
 
-            for arc in self.pyagrum_model.arcs():
-                parent = self.pyagrum_model.variable(arc[0]).name()
-                child = self.pyagrum_model.variable(arc[1]).name()
-                structure['edges'].append((parent, child))
+            structure['edges'] = list(self.pgmpy_model.edges())
 
-        elif self.pgmpy_model:
-            for node in self.pgmpy_model.nodes():
+        elif self.pyagrum_model and PYAGRUM_AVAILABLE:
+            for node in self.pyagrum_model.names():
                 category = self._get_variable_category(node)
                 structure['nodes'].append({
                     'name': node,
-                    'category': category
+                    'category': category,
+                    'domain': list(self.pyagrum_model.variable(node).labels())
                 })
                 structure['categories'][node] = category
 
-            structure['edges'] = list(self.pgmpy_model.edges())
+            for arc in self.pyagrum_model.arcs():
+                parent = self.pyagrum_model.variable(arc[0]).name()
+                child = self.pyagrum_model.variable(arc[1]).name()
+                structure['edges'].append((parent, child))
 
         return structure