Skip to content

Commit 6972062

Browse files
committed
fix: return the first feature of a structural frame
1 parent 5b34346 commit 6972062

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

LoopStructural/modelling/features/_structural_frame.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -134,7 +134,7 @@ def evaluate_value(self, pos, ignore_regions=False):
134134
v[:, 0] = self.features[0].evaluate_value(pos, ignore_regions=ignore_regions)
135135
v[:, 1] = self.features[1].evaluate_value(pos, ignore_regions=ignore_regions)
136136
v[:, 2] = self.features[2].evaluate_value(pos, ignore_regions=ignore_regions)
137-
return v
137+
return v[:,0]
138138

139139
def evaluate_gradient(self, pos, i=None, ignore_regions=False):
140140
"""
@@ -152,11 +152,7 @@ def evaluate_gradient(self, pos, i=None, ignore_regions=False):
152152
"""
153153
if i is not None:
154154
return self.features[i].support.evaluate_gradient(pos, ignore_regions=ignore_regions)
155-
return (
156-
self.features[0].support.evaluate_gradient(pos, ignore_regions=ignore_regions),
157-
self.features[1].support.evaluate_gradient(pos, ignore_regions=ignore_regions),
158-
self.features[2].support.evaluate_gradient(pos, ignore_regions=ignore_regions),
159-
)
155+
return self.features[0].support.evaluate_gradient(pos, ignore_regions=ignore_regions)
160156

161157
def get_data(self, value_map: Optional[dict] = None) -> List[Union[ValuePoints, VectorPoints]]:
162158
"""Return the data associated with the features in the

0 commit comments

Comments
 (0)