From 958364884c3605cef6bc1f535081795d216788d4 Mon Sep 17 00:00:00 2001 From: Derek Maggio Date: Tue, 16 Apr 2024 09:45:08 -0700 Subject: [PATCH] chore: update imports --- performance-metrics/src/performance_metrics/datashapes.py | 1 - .../src/performance_metrics/robot_context_tracker.py | 5 ++++- .../tests/performance_metrics/test_robot_context_tracker.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/performance-metrics/src/performance_metrics/datashapes.py b/performance-metrics/src/performance_metrics/datashapes.py index f0c9ceb9790..7743ed1723d 100644 --- 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 index 107719a6414..1c0294af080 100644 --- 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 index 4f615878895..2973f0d5990 100644 --- 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