Coverage for src/loman/consts.py: 100%

1"""Constants and enumerations for the loman computation engine.""" 

2 

3from enum import Enum 

4 

5 

6class States(Enum): 

7 """Possible states for a computation node.""" 

8 

9 PLACEHOLDER = 0 

10 UNINITIALIZED = 1 

11 STALE = 2 

12 COMPUTABLE = 3 

13 UPTODATE = 4 

14 ERROR = 5 

15 PINNED = 6 

16 

17 

18class NodeAttributes: 

19 """Constants for node attribute names in the computation graph.""" 

20 

21 VALUE = "value" 

22 STATE = "state" 

23 FUNC = "func" 

24 GROUP = "group" 

25 TAG = "tag" 

26 STYLE = "style" 

27 ARGS = "args" 

28 KWDS = "kwds" 

29 TIMING = "timing" 

30 EXECUTOR = "executor" 

31 CONVERTER = "converter" 

32 STORE = "store" 

33 

34 

35class EdgeAttributes: 

36 """Constants for edge attribute names in the computation graph.""" 

37 

38 PARAM = "param" 

39 

40 

41class SystemTags: 

42 """System-level tags used internally by loman.""" 

43 

44 SERIALIZE = "__serialize__" 

45 EXPANSION = "__expansion__" 

46 

47 

48class NodeTransformations: 

49 """Node transformation types for visualization.""" 

50 

51 CONTRACT = "contract" 

52 COLLAPSE = "collapse" 

53 EXPAND = "expand"