Skip to content

Commit 0a55972

Browse files
committed
[refactor] more compact tri-layer apply method
1 parent 12e6204 commit 0a55972

1 file changed

Lines changed: 4 additions & 10 deletions

File tree

src/layout.rs

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -343,16 +343,10 @@ struct TriLayer {
343343
}
344344

345345
impl TriLayer {
346-
fn apply(&self, last_layer: usize, second_to_last_layer: usize) -> Option<usize> {
347-
if (last_layer == self.activation_layers.0
348-
&& second_to_last_layer == self.activation_layers.1)
349-
|| (last_layer == self.activation_layers.1
350-
&& second_to_last_layer == self.activation_layers.0)
351-
{
352-
Some(self.target_layer)
353-
} else {
354-
None
355-
}
346+
fn apply(&self, layer_0: usize, layer_1: usize) -> Option<usize> {
347+
(self.activation_layers == (layer_0, layer_1)
348+
|| self.activation_layers == (layer_1, layer_0))
349+
.then_some(self.target_layer)
356350
}
357351
}
358352

0 commit comments

Comments
 (0)