Skip to content

Commit e79a58c

Browse files
committed
Codacy
1 parent eeaa0f5 commit e79a58c

File tree

3 files changed

+18
-36
lines changed

3 files changed

+18
-36
lines changed

pina/condition/data_condition.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -66,26 +66,22 @@ def _get_subclass(input, conditional_variables):
6666
)
6767
if is_tensor_input:
6868
return TensorDataCondition
69-
elif is_graph_input:
69+
if is_graph_input:
7070
return GraphDataCondition
71-
else:
72-
raise ValueError(
73-
"Invalid input types. "
74-
"Please provide either torch.Tensor or Graph objects."
75-
)
71+
72+
raise ValueError(
73+
"Invalid input types. "
74+
"Please provide either torch.Tensor or Graph objects."
75+
)
7676

7777

7878
class TensorDataCondition(DataCondition):
7979
"""
8080
DataCondition for torch.Tensor input data
8181
"""
8282

83-
pass
84-
8583

8684
class GraphDataCondition(DataCondition):
8785
"""
8886
DataCondition for Graph/Data input data
8987
"""
90-
91-
pass

pina/condition/input_equation_condition.py

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -60,26 +60,21 @@ def _get_subclass(input, equation):
6060
)
6161
if is_tensor_input:
6262
return InputTensorEquationCondition
63-
elif is_graph_input:
63+
if is_graph_input:
6464
return InputGraphEquationCondition
65-
else:
66-
raise ValueError(
67-
"Invalid input types. "
68-
"Please provide either torch.Tensor or Graph objects."
69-
)
65+
raise ValueError(
66+
"Invalid input types. "
67+
"Please provide either torch.Tensor or Graph objects."
68+
)
7069

7170

7271
class InputTensorEquationCondition(InputEquationCondition):
7372
"""
7473
InputEquationCondition subclass for torch.Tensor input data.
7574
"""
7675

77-
pass
78-
7976

8077
class InputGraphEquationCondition(InputEquationCondition):
8178
"""
8279
InputEquationCondition subclass for Graph input data.
8380
"""
84-
85-
pass

pina/condition/input_target_condition.py

Lines changed: 7 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -66,17 +66,16 @@ def _get_subclass(input, target):
6666

6767
if is_tensor_input and is_tensor_target:
6868
return TensorInputTensorTargetCondition
69-
elif is_tensor_input and is_graph_target:
69+
if is_tensor_input and is_graph_target:
7070
return TensorInputGraphTargetCondition
71-
elif is_graph_input and is_tensor_target:
71+
if is_graph_input and is_tensor_target:
7272
return GraphInputTensorTargetCondition
73-
elif is_graph_input and is_graph_target:
73+
if is_graph_input and is_graph_target:
7474
return GraphInputGraphTargetCondition
75-
else:
76-
raise ValueError(
77-
"Invalid input and target types. "
78-
"Please provide either torch.Tensor or Graph objects."
79-
)
75+
raise ValueError(
76+
"Invalid input and target types. "
77+
"Please provide either torch.Tensor or Graph objects."
78+
)
8079

8180
def __init__(self, input, target):
8281
"""
@@ -92,30 +91,22 @@ class TensorInputTensorTargetCondition(InputTargetCondition):
9291
InputTargetCondition subclass for torch.Tensor input and target data.
9392
"""
9493

95-
pass
96-
9794

9895
class TensorInputGraphTargetCondition(InputTargetCondition):
9996
"""
10097
InputTargetCondition subclass for torch.Tensor input and Graph/Data target
10198
data.
10299
"""
103100

104-
pass
105-
106101

107102
class GraphInputTensorTargetCondition(InputTargetCondition):
108103
"""
109104
InputTargetCondition subclass for Graph/Data input and torch.Tensor target
110105
data.
111106
"""
112107

113-
pass
114-
115108

116109
class GraphInputGraphTargetCondition(InputTargetCondition):
117110
"""
118111
InputTargetCondition subclass for Graph/Data input and target data.
119112
"""
120-
121-
pass

0 commit comments

Comments
 (0)