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

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

2 

3from importlib.metadata import PackageNotFoundError, version 

4 

5from invariant.async_executor import AsyncExecutor 

6from invariant.executor import Executor 

7from invariant.graph import Graph, GraphResolver, GraphVertex 

8from invariant.graph_serialization import ( 

9 GRAPH_DATA_URI_PREFIX, 

10 GRAPH_MEDIA_TYPE, 

11 dump_graph, 

12 dump_graph_data_uri, 

13 dump_graph_to_dict, 

14 dump_value_to_jsonable, 

15 graph_data_uri_cache_key, 

16 load_graph, 

17 load_graph_data_uri, 

18 load_graph_document, 

19 load_graph_document_from_dict, 

20 load_graph_from_dict, 

21 load_value_from_jsonable, 

22) 

23from invariant.node import Node, SubGraphNode, SwitchNode 

24from invariant.params import cel, ref 

25from invariant.registry import OpBinding, OpRegistry 

26from invariant.scheduler import ( 

27 InlineScheduler, 

28 InvocationRequest, 

29 ProcessPoolScheduler, 

30 RoutingScheduler, 

31 ThreadPoolScheduler, 

32) 

33from invariant.traits import OpTrait, op_traits 

34from invariant.yaml_serialization import load_graph_document_yaml, load_graph_yaml 

35 

36try: 

37 __version__ = version("invariant-core") 

38except PackageNotFoundError: 

39 __version__ = "0.0.0+unknown" 

40 

41__all__ = [ 

42 "AsyncExecutor", 

43 "Executor", 

44 "Graph", 

45 "GraphResolver", 

46 "GraphVertex", 

47 "GRAPH_DATA_URI_PREFIX", 

48 "GRAPH_MEDIA_TYPE", 

49 "InlineScheduler", 

50 "InvocationRequest", 

51 "Node", 

52 "OpBinding", 

53 "OpRegistry", 

54 "OpTrait", 

55 "ProcessPoolScheduler", 

56 "RoutingScheduler", 

57 "SubGraphNode", 

58 "SwitchNode", 

59 "ThreadPoolScheduler", 

60 "cel", 

61 "dump_graph", 

62 "dump_graph_data_uri", 

63 "dump_graph_to_dict", 

64 "dump_value_to_jsonable", 

65 "graph_data_uri_cache_key", 

66 "load_graph", 

67 "load_graph_data_uri", 

68 "load_graph_document", 

69 "load_graph_document_from_dict", 

70 "load_graph_document_yaml", 

71 "load_graph_from_dict", 

72 "load_graph_yaml", 

73 "load_value_from_jsonable", 

74 "op_traits", 

75 "ref", 

76 "__version__", 

77]