folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Formatting

Browse Source 
* Adding black as dev dependency
* Formatting pina code
* Formatting tests
This commit is contained in:
Dario Coscia
2025-02-24 11:26:49 +01:00
committed by Nicola Demo
parent 4c4482b155
commit 42ab1a666b
77 changed files with 1170 additions and 924 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
pina/condition/input_output_condition.py
View File

@@ -24,8 +24,11 @@ class InputOutputPointsCondition(ConditionInterface):
self.output_points = output_points
def __setattr__(self, key, value):
if (key == 'input_points') or (key == 'output_points'):
check_consistency(value, (LabelTensor, Graph, torch.Tensor, torch_geometric.data.Data))
if (key == "input_points") or (key == "output_points"):
check_consistency(
value,
(LabelTensor, Graph, torch.Tensor, torch_geometric.data.Data),
)
InputOutputPointsCondition.__dict__[key].__set__(self, value)
elif key in ('_problem', '_condition_type'):
elif key in ("_problem", "_condition_type"):
super().__setattr__(key, value)