Handle YOLO model loading issue caused by latest torch release (#100)

* Update yolo dependency

* Re-enable all tests

Author: zhong-al

pyproject.toml

@@ -19,7 +19,7 @@ authors = [
 ]
 description = "Tools for working with data for annotating animal behavior. These were specifically designed during construction of the KABR dataset."
 readme = "README.md"
-requires-python = ">=3.10, <3.11"
+requires-python = ">=3.10, <3.12"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: MIT License",

requirements.txt

@@ -6,7 +6,7 @@ tqdm>=4.64.1
 torch>=1.10.0
 natsort>=8.2.0
 ruamel.yaml>=0.17.21
-ultralytics~=8.0.36
+ultralytics>=8.0.36
 pandas>=1.3.5
 pillow==10.4.0
 scikit-learn==1.5.1

src/kabr_tools/utils/yolo.py

@@ -1,7 +1,6 @@
 import numpy as np
 from ultralytics import YOLO
 
-
 class YOLOv8:
     def __init__(self, weights="yolov8x.pt", imgsz=640, conf=0.5):
         self.conf = conf