Skip to content

Commit

Permalink
Remove out-of-order logic
Browse files Browse the repository at this point in the history
  • Loading branch information
amyeroberts committed Nov 28, 2023
1 parent 1db3de3 commit f5b5db4
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions tests/test_backbone_common.py
Original file line number Diff line number Diff line change
Expand Up @@ -218,12 +218,6 @@ def test_backbone_stage_selection(self):
self.assertIsInstance(outputs.feature_maps, tuple)
self.assertTrue(len(outputs.feature_maps) == 2)

# Test out-of-order feature maps returned in passed-in order
config.out_indices = [2, 1]
backbone = backbone_class(config)
backbone.to(torch_device)
backbone.eval()

# Order of channels returned is same as order of channels iterating over stage names
channels_from_stage_names = [
backbone.out_feature_channels[name] for name in backbone.stage_names if name in backbone.out_features
Expand Down

0 comments on commit f5b5db4

Please sign in to comment.