From 480487651e764ff7bf38aab7f02c8da8d4d043e7 Mon Sep 17 00:00:00 2001 From: Jason Jafari Date: Wed, 19 Jun 2024 22:40:37 -0400 Subject: [PATCH] chore: bump version to 1.0.31 --- package.json | 2 +- pytests/test_mlModelSaver.py | 7 ++++++- setup.py | 2 +- 3 files changed, 8 insertions(+), 3 deletions(-) diff --git a/package.json b/package.json index 194ef5a..767852a 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "mlModelSaver", - "version": "1.0.30", + "version": "1.0.31", "description": "Make life easier for save and serving ml models", "main": "index.js", "repository": "git@github.com:smartdev-ca/mlModelSaver.git", diff --git a/pytests/test_mlModelSaver.py b/pytests/test_mlModelSaver.py index b396712..46cfeaf 100644 --- a/pytests/test_mlModelSaver.py +++ b/pytests/test_mlModelSaver.py @@ -23,7 +23,12 @@ def test_ensureCLassInstance(): assert mlModelSaverInstance1.baseRelativePath == "test_baseRelativePath" assert mlModelSaverInstance1.modelsFolder == "test_baseRelativePath/test_modelsFolder" tesSupportedModels = mlModelSaverInstance1.showSupportedModels() - assert tesSupportedModels == ['sm.OLS', 'sm.Logit', 'sklearn.neighbors.KNeighborsClassifier'] + assert tesSupportedModels == [ + 'sm.OLS', + 'sm.Logit', + 'sklearn.neighbors.KNeighborsClassifier', + 'sklearn.tree.DecisionTreeClassifier"' + ] def test_OLS_LinearRegression(): diff --git a/setup.py b/setup.py index 62f2cc0..1ca67d7 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name='mlModelSaver', - version='1.0.30', + version='1.0.31', packages=find_packages(), description='Make life easier for saving and serving ML models', long_description=open('DOCS.md').read(), # Assumes you have a README.md file