export TiramisuServer from tiramisu modeule

Author: skourta

tiralib/tiramisu/__init__.py

@@ -7,6 +7,7 @@
 from .tiramisu_iterator_node import IteratorIdentifier, IteratorNode
 from .tiramisu_program import TiramisuProgram
 from .tiramisu_tree import TiramisuTree
+from .function_server import FunctionServer
 
 __all__ = [
     "CompilingService",
@@ -16,4 +17,5 @@
     "IteratorNode",
     "tiramisu_actions",
     "IteratorIdentifier",
+    "FunctionServer",
 ]

tiralib/tiramisu/tiramisu_program.py

@@ -170,7 +170,7 @@ def init_server(
         load_isl_ast=False,
         load_tree=False,
         reuse_server=False,
-    ):
+    ) -> "TiramisuProgram":
         # Initiate an instante of the TiramisuProgram class
         tiramisu_prog = cls()
         tiramisu_prog.cpp_code = cpp_code