pythonflow: declare class vars inside __init__

utahplt · Jul 21, 2023 · b1cb10b · b1cb10b
1 parent 4f4a0a2
commit b1cb10b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Benchmark/pythonflow/shallow/main.py b/Benchmark/pythonflow/shallow/main.py
@@ -16,6 +16,8 @@ def __init__(self: PythonFlow) -> None:
         self.residual = []
         self.total_flow = 0
         self.file_name = "graph2.txt"
+        self.options = []
+        self.cost = []
 
         self.load_file()
         self.init_flow()

diff --git a/Benchmark/pythonflow/untyped/main.py b/Benchmark/pythonflow/untyped/main.py
@@ -29,6 +29,8 @@ def __init__(self):
         self.total_flow = 0
         # file name
         self.file_name = "graph2.txt"
+        self.options = []
+        self.cost = []
 
         """  -- testing --
         n = 20
@@ -183,4 +185,4 @@ def print_graph(self, graph):
 flow = PythonFlow()
 
 
-flow.main_algorithm()
+flow.main_algorithm()