Coverage for neuralfields/__init__.py: 100%

1from pathlib import Path 

2 

3from dunamai import Version 

4 

5from neuralfields.custom_layers import IndependentNonlinearitiesLayer, MirroredConv1d, init_param_ 

6from neuralfields.custom_types import ActivationFunction, PotentialsDynamicsType 

7from neuralfields.neural_fields import NeuralField 

8from neuralfields.potential_based import PotentialBased 

9from neuralfields.simple_neural_fields import ( 

10 SimpleNeuralField, 

11 pd_capacity_21, 

12 pd_capacity_21_abs, 

13 pd_capacity_32, 

14 pd_capacity_32_abs, 

15 pd_cubic, 

16 pd_linear, 

17) 

18 

19 

20# Define variables for important folders. 

21SRC_DIR = Path(__file__).resolve().parent 

22EXAMPLES_DIR = SRC_DIR.parent / "examples" 

23 

24# Get the project's version dynamically. 

25VERSION = Version.from_git().base 

26 

27# Set the public API. 

28__all__ = [ 

29 "EXAMPLES_DIR", 

30 "SRC_DIR", 

31 "VERSION", 

32 "ActivationFunction", 

33 "IndependentNonlinearitiesLayer", 

34 "MirroredConv1d", 

35 "NeuralField", 

36 "PotentialBased", 

37 "PotentialsDynamicsType", 

38 "SimpleNeuralField", 

39 "init_param_", 

40 "pd_capacity_21", 

41 "pd_capacity_21_abs", 

42 "pd_capacity_32", 

43 "pd_capacity_32_abs", 

44 "pd_cubic", 

45 "pd_linear", 

46]