reformat

Author: Howardhuang98

LearningWithExpertKnowledge/estimator.py

@@ -1,11 +1,13 @@
+import logging
+from collections import deque
 from itertools import permutations
-from LearningWithExpertKnowledge.expert import *
-from LearningWithExpertKnowledge.graph import DAG
+
 import networkx as nx
 import numpy as np
 from tqdm import trange
-from collections import deque
-import logging
+
+from LearningWithExpertKnowledge.expert import *
+from LearningWithExpertKnowledge.graph import DAG
 
 
 class Estimator:

LearningWithExpertKnowledge/graph.py

@@ -1,5 +1,5 @@
-import networkx as nx
 import matplotlib.pyplot as plt
+import networkx as nx
 
 
 class DAG(nx.DiGraph):
@@ -11,11 +11,6 @@ def __init__(self, edges=None):
  cycles = list(nx.find_cycle(self))
  except nx.NetworkXNoCycle:
  pass
- else:
- out_str = "Cycles are not allowed in a DAG."
- out_str += "\nEdges indicating the path taken for a loop: "
- out_str += "".join([f"({u},{v}) " for (u, v) in cycles])
- raise ValueError(out_str)
 
  def get_parents(self, node):
  """

LearningWithExpertKnowledge/run/run0.py

@@ -1,7 +1,5 @@
 from LearningWithExpertKnowledge.estimator import *
 from LearningWithExpertKnowledge.expert import *
-from LearningWithExpertKnowledge.graph import *
-import pandas as pd
 
 if __name__ == '__main__':
  """