Fix imports

onnx · Aug 12, 2024 · 6bf334a · 6bf334a
1 parent 468ebb7
commit 6bf334a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/turnkeyml/run/basert.py b/src/turnkeyml/run/basert.py
@@ -7,7 +7,7 @@
 from abc import ABC, abstractmethod
 import torch
 import numpy as np
-from turnkeyml.common.performance import MeasuredPerformance, Device
+from turnkeyml.run.performance import MeasuredPerformance, Device
 import turnkeyml.common.build as build
 import turnkeyml.common.exceptions as exp
 import turnkeyml.common.filesystem as fs

diff --git a/src/turnkeyml/run/benchmark_model.py b/src/turnkeyml/run/benchmark_model.py
@@ -10,7 +10,7 @@
     apply_default_runtime,
 )
 import turnkeyml.cli.parser_helpers as parser_helpers
-from turnkeyml.common.performance import Device, parse_device
+from turnkeyml.run.performance import Device, parse_device
 
 default_iterations = 100
 benchmark_default_device = "x86"

diff --git a/src/turnkeyml/run/torchrt/runtime.py b/src/turnkeyml/run/torchrt/runtime.py
@@ -7,7 +7,7 @@
 import torch
 import numpy as np
 from turnkeyml.run.basert import BaseRT
-from turnkeyml.common.performance import MeasuredPerformance
+from turnkeyml.run.performance import MeasuredPerformance
 from turnkeyml.run.onnxrt.execute import get_cpu_specs
 import turnkeyml.common.build as build
 import turnkeyml.common.exceptions as exp