From 4a95a634bdc1bf284b8b4253a4ab40389ddbd6f4 Mon Sep 17 00:00:00 2001 From: Naveed Jooma Date: Wed, 18 Feb 2026 15:44:15 -0500 Subject: [PATCH] Add packaging is a dependency for MLModel --- pyproject.toml | 3 ++- src/viam/services/mlmodel/utils.py | 1 - tests/mocks/__init__.py | 1 - 3 files changed, 2 insertions(+), 3 deletions(-) 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