Coverage for src / invariant / __init__.py: 83.33%

1"""Invariant: A deterministic execution engine for DAGs.""" 

2 

3from importlib.metadata import PackageNotFoundError, version 

4 

5from invariant.executor import Executor 

6from invariant.graph import Graph, GraphResolver, GraphVertex 

7from invariant.graph_serialization import ( 

8 GRAPH_OUTPUT_DATA_URI_PREFIX, 

9 GRAPH_OUTPUT_MEDIA_TYPE, 

10 dump_graph, 

11 dump_graph_output_data_uri, 

12 dump_graph_output_to_dict, 

13 dump_graph_to_dict, 

14 dump_value_to_jsonable, 

15 load_graph, 

16 load_graph_from_dict, 

17 load_graph_output_data_uri, 

18 load_graph_output_from_dict, 

19 load_value_from_jsonable, 

20) 

21from invariant.node import Node, SubGraphNode 

22from invariant.params import cel, ref 

23from invariant.registry import OpRegistry 

24 

25try: 

26 __version__ = version("invariant-core") 

27except PackageNotFoundError: 

28 __version__ = "0.0.0+unknown" 

29 

30__all__ = [ 

31 "Executor", 

32 "Graph", 

33 "GraphResolver", 

34 "GraphVertex", 

35 "GRAPH_OUTPUT_DATA_URI_PREFIX", 

36 "GRAPH_OUTPUT_MEDIA_TYPE", 

37 "Node", 

38 "OpRegistry", 

39 "SubGraphNode", 

40 "cel", 

41 "dump_graph", 

42 "dump_graph_output_data_uri", 

43 "dump_graph_output_to_dict", 

44 "dump_graph_to_dict", 

45 "dump_value_to_jsonable", 

46 "load_graph", 

47 "load_graph_output_data_uri", 

48 "load_graph_output_from_dict", 

49 "load_graph_from_dict", 

50 "load_value_from_jsonable", 

51 "ref", 

52 "__version__", 

53]