update setup.py; removed unused imports

neuralmagic · Jul 24, 2023 · 497b1c0 · 497b1c0
1 parent 19ad588
commit 497b1c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -133,7 +133,7 @@ def _parse_requirements_file(file_path):
     "opencv-python<=4.6.0.66",
 ]
 _yolo_integration_deps = [
-    "torchvision>=0.3.0,<0.14",
+    "torchvision>=0.3.0,<=0.15.1",
     "opencv-python<=4.6.0.66",
 ]
 _openpifpaf_integration_deps = [

diff --git a/src/deepsparse/yolov8/utils/validation/deepsparse_validator.py b/src/deepsparse/yolov8/utils/validation/deepsparse_validator.py
@@ -10,9 +10,7 @@
 from deepsparse.yolov8.utils.validation.helpers import schema_to_tensor
 from ultralytics.yolo.data.utils import check_det_dataset
 from ultralytics.yolo.utils import (
-    DEFAULT_CFG,
     LOGGER,
-    SETTINGS,
     TQDM_BAR_FORMAT,
     callbacks,
 )
@@ -40,7 +38,7 @@ def __call__(self, classes: Dict[int, str]):
         # for validation when self.training is True
         callbacks.add_integration_callbacks(self)  # TODO: Missing inputs?
         self.run_callbacks("on_val_start")
-        self.device = select_device(self.args.device, self.args.batch)  # TODO: Fix
+        self.device = select_device(self.args.device, self.args.batch)
         # self.args.half &= self.device.type != "cpu" # TODO: not commented out here?
         self.data = check_det_dataset(self.args.data)
         if isinstance(self.data["path"], str):