physicsnemo/.importlinter at fff32a557f2ab98e02807a85e9091e5c159c87ab · peterdsharpe/physicsnemo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
[importlinter]
root_package = physicsnemo
include_external_packages = True
contract_types =
    forbidden_import: test.ci_tests.prevent_untracked_imports.ForbiddenImportContract
exclude_type_checking_imports = true

[importlinter:contract:physicsnemo-modules]
name = Prevent Upward Imports in the PhysicsNemo Structure
type = layers
containers=
   physicsnemo
layers =
   experimental
   active_learning : diffusion
   models : datapipes : metrics : domain_parallel
   nn
   utils
   distributed
   core

[importlinter:contract:physicsnemo-core]
name = Control Dependencies in PhysicsNeMo core
type = layers
containers=
   physicsnemo.core
layers =
   module : registry
   meta
   warnings | version_check | filesystem


[importlinter:contract:physicsnemo-distributed]
name = Control Dependencies in PhysicsNeMo distributed
type = layers
containers=
   physicsnemo.distributed
layers =
   fft | autograd
   mappings
   utils
   manager
   config

[importlinter:contract:physicsnemo-utils]
name = Control Dependencies in PhysicsNeMo utils
type = layers
containers=
   physicsnemo.utils
layers =
   mesh | insolation | zenith_angle
   profiling
   checkpoint
   capture
   logging | memory

[importlinter:contract:physicsnemo-nn]
name = Control Dependencies in PhysicsNeMo nn
type = layers
containers=
   physicsnemo.nn
layers =
   module.fourier_layers | module.transformer_layers | module.unet_layers
   module.dgm_layers | module.mlp_layers | module.fully_connected_layers | module.gnn_layers | module.attention_layers
   module.activations | module.ball_query | module.conv_layers  | module.drop | module.fft  | module.fused_silu | module.interpolation | module.kan_layers  | module.resample_layers | module.siren_layers | module.spectral_layers | module.transformer_decoder  | module.weight_fact | module.weight_norm | module.embedding_layers | module.group_norm | module.hpx
   functional
   module.utils

[importlinter:contract:physicsnemo-nn-gnn-layers]
name = Control Internal Dependencies in PhysicsNeMo nn GNN Layers
type = layers
containers=
   physicsnemo.nn.module.gnn_layers
layers =
   bsms
   mesh_graph_decoder | mesh_graph_encoder
   mesh_node_block | mesh_edge_block
   mesh_graph_mlp
   utils
   graph
   distributed_graph
   graph_types

[importlinter:contract:physicsnemo-models]
name = Prevent Imports between physicsnemo models
type = layers
containers=
   physicsnemo.models
layers =
   mesh_reduced
   afno | dlwp | dlwp_healpix | domino | dpot | fengwu | figconvnet | fno | graphcast | meshgraphnet | pangu | pix2pix | rnn |  srrn | swinvrnn | topodiff | transolver  | vfgn | diffusion_unets
   unet

[importlinter:contract:physicsnemo-diffusion]
name = Control Internal Dependencies in PhysicsNeMo diffusion
type = layers
containers=
   physicsnemo.diffusion
layers =
   generate
   samplers : metrics
   noise_schedulers | multi_diffusion | preconditioners | denoisers
   utils

[importlinter:contract:physicsnemo-external-imports]
name = Prevent Non-listed external imports in physicsnemo
type = forbidden_import
container = physicsnemo
exclude =
   datapipes