Use alternative function already in koalas

adamburkegh · Mar 5, 2024 · 061daf7 · 061daf7
1 parent 47fd40b
commit 061daf7
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/cgedq/mine.py b/cgedq/mine.py
@@ -144,7 +144,7 @@ def filterByTimeOnDate(sslog:set, years: int ) -> set:
 
 
 def exportNetToImage(vard,oname,pn):
-    dotStr = exportToDOT(pn)
+    dotStr = convert_net_to_dot(pn)
     export_DOT_to_image(vard,oname,dotStr)
 
 def mineByTime(vard,fname,sslog:set,years:int,noise=0.0,font=None):
@@ -246,13 +246,6 @@ def main():
     info("Done")
 
 
-#
-# TODO Push up into koalas models/petrinet
-#
-
-def exportToDOT(net:LabelledPetriNet) -> str:
-    return convert_net_to_dot(PetriNetDOTFormatter(net))
-
 if __name__ == '__main__':
     main()