diff --git a/pyproject.toml b/pyproject.toml
index 2eebe33658..b3916766f6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -44,7 +44,8 @@ Repository = "https://github.com/viamrobotics/viam-python-sdk"
 
 [project.optional-dependencies]
 mlmodel = [
-	"numpy"
+	"numpy",
+	"packaging",
 ]
 
 [build-system]
diff --git a/src/viam/services/mlmodel/utils.py b/src/viam/services/mlmodel/utils.py
index c2dbead820..37ba3022da 100644
--- a/src/viam/services/mlmodel/utils.py
+++ b/src/viam/services/mlmodel/utils.py
@@ -44,7 +44,6 @@ def make_ndarray(flat_data, dtype, shape):
         # Creating our array as a uint32 array initially and then casting to int16 solves this.
         if Version(np.__version__) >= Version("2") and dtype == np.int16:
             arr = np.astype(make_array(flat_data, np.uint32), np.int16)  # pyright: ignore [reportAttributeAccessIssue]
-
         else:
             arr = make_array(flat_data, dtype)
         return arr.reshape(shape)
diff --git a/tests/mocks/__init__.py b/tests/mocks/__init__.py
index 803a5935cd..719e976ab6 100644
--- a/tests/mocks/__init__.py
+++ b/tests/mocks/__init__.py
@@ -1,5 +1,4 @@
 import inspect
-from typing import Dict, List, Sequence, Set, Tuple, Union, get_args, get_origin, get_type_hints
 from unittest.mock import AsyncMock, create_autospec