Fix Codacy Warnings (#477)
--------- Co-authored-by: Dario Coscia <dariocos99@gmail.com>
This commit is contained in:
committed by
Nicola Demo
parent
e3790e049a
commit
4177bfbb50
@@ -1,8 +1,8 @@
|
||||
"""Module Averaging Neural Operator."""
|
||||
|
||||
import torch
|
||||
from torch import nn, cat
|
||||
from .block import AVNOBlock
|
||||
from torch import nn
|
||||
from .block.average_neural_operator_block import AVNOBlock
|
||||
from .kernel_neural_operator import KernelNeuralOperator
|
||||
from ..utils import check_consistency
|
||||
|
||||
@@ -110,9 +110,9 @@ class AveragingNeuralOperator(KernelNeuralOperator):
|
||||
"""
|
||||
points_tmp = x.extract(self.coordinates_indices)
|
||||
new_batch = x.extract(self.field_indices)
|
||||
new_batch = cat((new_batch, points_tmp), dim=-1)
|
||||
new_batch = torch.cat((new_batch, points_tmp), dim=-1)
|
||||
new_batch = self._lifting_operator(new_batch)
|
||||
new_batch = self._integral_kernels(new_batch)
|
||||
new_batch = cat((new_batch, points_tmp), dim=-1)
|
||||
new_batch = torch.cat((new_batch, points_tmp), dim=-1)
|
||||
new_batch = self._projection_operator(new_batch)
|
||||
return new_batch
|
||||
|
||||
Reference in New Issue
Block a user