chore: update imports

Opentrons · Apr 16, 2024 · 9583648 · 9583648
1 parent f6ce258
commit 9583648
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/performance-metrics/src/performance_metrics/datashapes.py b/performance-metrics/src/performance_metrics/datashapes.py
@@ -1,6 +1,5 @@
 """Defines data classes and enums used in the performance metrics module."""
 
-from enum import Enum
 import dataclasses
 from typing import Tuple
 from opentrons_shared_data.performance.dev_types import RobotContextState

diff --git a/performance-metrics/src/performance_metrics/robot_context_tracker.py b/performance-metrics/src/performance_metrics/robot_context_tracker.py
@@ -26,7 +26,10 @@
 from performance_metrics.datashapes import (
     RawContextData,
 )
-from opentrons_shared_data.performance.dev_types import RobotContextState, SupportsTracking
+from opentrons_shared_data.performance.dev_types import (
+    RobotContextState,
+    SupportsTracking,
+)
 
 P = ParamSpec("P")
 R = TypeVar("R")

diff --git a/performance-metrics/tests/performance_metrics/test_robot_context_tracker.py b/performance-metrics/tests/performance_metrics/test_robot_context_tracker.py
@@ -4,7 +4,7 @@
 from pathlib import Path
 import pytest
 from performance_metrics.robot_context_tracker import RobotContextTracker
-from performance_metrics.datashapes import RobotContextState
+from opentrons_shared_data.performance.dev_types import RobotContextState
 from time import sleep
 
 # Corrected times in seconds