From 54300fb26dfc9962aa7193ba6943172ed919c802 Mon Sep 17 00:00:00 2001 From: nlathia Date: Sat, 21 Feb 2026 12:48:27 +0000 Subject: [PATCH] Fix pkg_resources migration issues - Import PackageNotFoundError in __init__.py to fix NameError - Remove platform-specific np.float96 from model_manager.py Co-Authored-By: Claude Haiku 4.5 --- modelstore/__init__.py | 2 +- modelstore/models/model_manager.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/modelstore/__init__.py b/modelstore/__init__.py index 87324397..399d457c 100644 --- a/modelstore/__init__.py +++ b/modelstore/__init__.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from importlib.metadata import version +from importlib.metadata import PackageNotFoundError, version # pylint: disable=unused-import from modelstore.model_store import ModelStore diff --git a/modelstore/models/model_manager.py b/modelstore/models/model_manager.py index 443e6124..a4275ac5 100644 --- a/modelstore/models/model_manager.py +++ b/modelstore/models/model_manager.py @@ -225,7 +225,7 @@ def upload( def _format_numpy(model_params: dict) -> dict: for key, value in model_params.items(): - if isinstance(value, (np.float16, np.float32, np.float64, np.float96)): + if isinstance(value, (np.float16, np.float32, np.float64)): model_params[key] = float(value) if isinstance(value, np.ndarray): model_params[key] = value.tolist()