Correct bug in draw lib

bbodin · Oct 31, 2018 · c5a8597 · c5a8597
1 parent 3287a3b
commit c5a8597
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/Turbine/draw/draw_dot.py b/Turbine/draw/draw_dot.py
@@ -58,9 +58,10 @@ def write_jpeg(self, name=None):
         self._dotGraph.write_jpeg(name + '.jpeg')
 
 
-file_path_bs = "simpl_BlackScholes.tur"
-# file_path_jpeg = "../../experimentations/indus/JPEG2000.tur"
-dataflow = read_tur_file(file_path_bs)
-# dataflow = get_csdf_into_sdf(dataflow)
-d = Dot(dataflow)
-d.write_pdf("test")
+if __name__ == "__main__":
+    file_path_bs = "simpl_BlackScholes.tur"
+    # file_path_jpeg = "../../experimentations/indus/JPEG2000.tur"
+    dataflow = read_tur_file(file_path_bs)
+    # dataflow = get_csdf_into_sdf(dataflow)
+    d = Dot(dataflow)
+    d.write_pdf("test")