🎨 Format Python code with psf/black

This commit is contained in:
ndem0
2024-04-02 08:24:40 +00:00
committed by Nicola Demo
parent 766856494a
commit 14dfa62402
5 changed files with 40 additions and 29 deletions

View File

@@ -41,7 +41,7 @@ class LowRankNeuralOperator(KernelNeuralOperator):
inner_size=20,
n_layers=2,
func=torch.nn.Tanh,
bias=True
bias=True,
):
"""
:param torch.nn.Module lifting_net: The neural network for lifting
@@ -105,13 +105,18 @@ class LowRankNeuralOperator(KernelNeuralOperator):
self.coordinates_indices = coordinates_indices
self.field_indices = field_indices
integral_net = nn.Sequential(
*[LowRankBlock(input_dimensions=len(coordinates_indices),
embedding_dimenion=output_lifting_net,
rank=rank,
inner_size=inner_size,
n_layers=n_layers,
func=func,
bias=bias) for _ in range(n_kernel_layers)]
*[
LowRankBlock(
input_dimensions=len(coordinates_indices),
embedding_dimenion=output_lifting_net,
rank=rank,
inner_size=inner_size,
n_layers=n_layers,
func=func,
bias=bias,
)
for _ in range(n_kernel_layers)
]
)
super().__init__(lifting_net, integral_net, projecting_net)