replace missing import

Flagsmith · Nov 21, 2023 · 3bd2564 · 3bd2564
1 parent 5a908b0
commit 3bd2564
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/api/environments/identities/tests/test_models.py b/api/environments/identities/tests/test_models.py
@@ -1,6 +1,13 @@
 import pytest
 from core.constants import FLOAT
 from django.utils import timezone
+from flag_engine.segments.constants import (
+    EQUAL,
+    GREATER_THAN,
+    GREATER_THAN_INCLUSIVE,
+    LESS_THAN_INCLUSIVE,
+    NOT_EQUAL,
+)
 from rest_framework.test import APITestCase
 
 from environments.identities.models import Identity
@@ -15,16 +22,7 @@
 from features.value_types import BOOLEAN, INTEGER, STRING
 from organisations.models import Organisation
 from projects.models import Project
-from segments.models import (
-    EQUAL,
-    GREATER_THAN,
-    GREATER_THAN_INCLUSIVE,
-    LESS_THAN_INCLUSIVE,
-    NOT_EQUAL,
-    Condition,
-    Segment,
-    SegmentRule,
-)
+from segments.models import Condition, Segment, SegmentRule
 
 from .helpers import (
     create_trait_for_identity,