Fix NameError: name 'ScheduleLineElement' is not defined error in Arg…

…os (#500) * Fix NameError: name 'ScheduleLineElement' is not defined error in Argos * Remove unused import to fix Python linter errors
sparcians · May 3, 2024 · 3092846 · 3092846
1 parent 2f3dc4c
commit 3092846
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/helios/pipeViewer/pipe_view/model/element_value.py b/helios/pipeViewer/pipe_view/model/element_value.py
@@ -4,13 +4,11 @@
 # [Element, val].
 from __future__ import annotations
 from bisect import bisect, bisect_left
-from typing import Any, Dict, List, Optional, Tuple, Union, TYPE_CHECKING
+from typing import Any, Dict, List, Optional, Tuple, Union
 from . import content_options as content
 from . import highlighting_utils
 from .element import Element, FakeElement, PropertyValue
-
-if TYPE_CHECKING:
-    from .schedule_element import ScheduleLineElement
+from .schedule_element import ScheduleLineElement
 
 
 TimedVal = Tuple[Optional[Union[int, str]], Tuple[int, int]]

diff --git a/helios/pipeViewer/pipe_view/model/schedule_element.py b/helios/pipeViewer/pipe_view/model/schedule_element.py
@@ -16,7 +16,9 @@
                       PropertyValue,
                       ValidatedPropertyDict)
 from . import element_propsvalid as valid
-from .element_value import Element_Value, FakeElementValue
+
+if TYPE_CHECKING:
+    from .element_value import Element_Value, FakeElementValue
 
 if TYPE_CHECKING:
     from .clock_manager import ClockManager
@@ -387,6 +389,8 @@ def GetQueryFrame(self, period: int) -> Tuple[int, int]:
     def DetectCollision(self,
                         pt: Union[Tuple[int, int], wx.Point],
                         pair: Element_Value) -> FakeElementValue:
+        from .element_value import FakeElementValue
+
         mx, my = pt
 
         period = pair.GetClockPeriod()