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

1"""Loman: A Python library for building computation graphs. 

2 

3Loman provides tools for creating and managing dependency-aware computation graphs 

4where nodes represent data or calculations, and edges represent dependencies. 

5""" 

6 

7import loman.util as util 

8import loman.visualization as viz 

9from loman.computeengine import ( 

10 C, 

11 Computation, 

12 ComputationEvent, 

13 block, 

14 calc_node, 

15 computation_factory, 

16 input_node, 

17 node, 

18 repeated_blocks, 

19) 

20from loman.consts import NodeTransformations, States 

21from loman.exception import ( 

22 CannotInsertToPlaceholderNodeError, 

23 DeserializedError, 

24 FittingError, 

25 InvalidBlockTypeError, 

26 LoopDetectedError, 

27 MapError, 

28 NonExistentNodeError, 

29 SerializationError, 

30 ValidationError, 

31) 

32from loman.nodekey import Name, Names, NodeKey, to_nodekey 

33from loman.planning import ExecutionPlan, ValidationReport 

34from loman.serialization import ComputationSerializer, SerializationProfile 

35from loman.util import BlockContext, BlockFeature, FanIn, FanOut, IdNode, InputValue, PlannedNode, Positional 

36from loman.visualization import GraphView 

37 

38# Backward compatibility alias 

39ComputationFactory = computation_factory 

40 

41__all__ = [ 

42 "BlockContext", 

43 "BlockFeature", 

44 "C", 

45 "CannotInsertToPlaceholderNodeError", 

46 "Computation", 

47 "ComputationEvent", 

48 "ComputationFactory", # Backward compatibility 

49 "ComputationSerializer", 

50 "DeserializedError", 

51 "ExecutionPlan", 

52 "FanIn", 

53 "FanOut", 

54 "FittingError", 

55 "GraphView", 

56 "IdNode", 

57 "InputValue", 

58 "InvalidBlockTypeError", 

59 "LoopDetectedError", 

60 "MapError", 

61 "Name", 

62 "Names", 

63 "NodeKey", 

64 "NodeTransformations", 

65 "NonExistentNodeError", 

66 "PlannedNode", 

67 "Positional", 

68 "SerializationError", 

69 "SerializationProfile", 

70 "States", 

71 "ValidationError", 

72 "ValidationReport", 

73 "block", 

74 "calc_node", 

75 "computation_factory", 

76 "input_node", 

77 "node", 

78 "repeated_blocks", 

79 "to_nodekey", 

80 "util", 

81 "viz", 

82]