Skip to content

Commit 7c2b321

Browse files
author
Tim Huff
committed
fixing a linter error
1 parent 5ca292e commit 7c2b321

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

test/unit/test_edge_config.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import pytest
2+
23
from groundlight.edge import (
34
DEFAULT,
45
EDGE_ANSWERS_WITH_ESCALATION,
@@ -8,6 +9,9 @@
89
InferenceConfig,
910
)
1011

12+
ONE_DETECTOR = 1
13+
TWO_DETECTORS = 2
14+
1115

1216
def test_edge_endpoint_config_is_not_subclass_of_detectors_config():
1317
assert not issubclass(EdgeEndpointConfig, DetectorsConfig)
@@ -32,7 +36,7 @@ def test_add_detector_allows_equivalent_named_inference_config():
3236
),
3337
)
3438

35-
assert len(detectors_config.detectors) == 2
39+
assert len(detectors_config.detectors) == TWO_DETECTORS
3640
assert list(detectors_config.edge_inference_configs.keys()) == ["custom_config"]
3741

3842

@@ -64,8 +68,8 @@ def test_from_detectors_config_copies_detector_data():
6468
config = EdgeEndpointConfig.from_detectors_config(detectors_config)
6569
detectors_config.add_detector("det_2", DEFAULT)
6670

67-
assert len(config.detectors) == 1
68-
assert len(detectors_config.detectors) == 2
71+
assert len(config.detectors) == ONE_DETECTOR
72+
assert len(detectors_config.detectors) == TWO_DETECTORS
6973

7074

7175
def test_inference_config_validation_errors():

0 commit comments

Comments
 (0)