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

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_DATA_URI_PREFIX, 

9 GRAPH_MEDIA_TYPE, 

10 dump_graph, 

11 dump_graph_data_uri, 

12 dump_graph_to_dict, 

13 dump_value_to_jsonable, 

14 graph_data_uri_cache_key, 

15 load_graph, 

16 load_graph_data_uri, 

17 load_graph_document, 

18 load_graph_document_from_dict, 

19 load_graph_from_dict, 

20 load_value_from_jsonable, 

21) 

22from invariant.node import Node, SubGraphNode, SwitchNode 

23from invariant.params import cel, ref 

24from invariant.registry import OpRegistry 

25from invariant.yaml_serialization import load_graph_document_yaml, load_graph_yaml 

26 

27try: 

28 __version__ = version("invariant-core") 

29except PackageNotFoundError: 

30 __version__ = "0.0.0+unknown" 

31 

32__all__ = [ 

33 "Executor", 

34 "Graph", 

35 "GraphResolver", 

36 "GraphVertex", 

37 "GRAPH_DATA_URI_PREFIX", 

38 "GRAPH_MEDIA_TYPE", 

39 "Node", 

40 "OpRegistry", 

41 "SubGraphNode", 

42 "SwitchNode", 

43 "cel", 

44 "dump_graph", 

45 "dump_graph_data_uri", 

46 "dump_graph_to_dict", 

47 "dump_value_to_jsonable", 

48 "graph_data_uri_cache_key", 

49 "load_graph", 

50 "load_graph_data_uri", 

51 "load_graph_document", 

52 "load_graph_document_from_dict", 

53 "load_graph_document_yaml", 

54 "load_graph_from_dict", 

55 "load_graph_yaml", 

56 "load_value_from_jsonable", 

57 "ref", 

58 "__version__", 

59]