diff --git a/package.json b/package.json index f1cef28..bdd7eb3 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "mlModelSaver", - "version": "1.0.17", + "version": "1.0.18", "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 13ac61c..357f0ac 100644 --- a/pytests/test_mlModelSaver.py +++ b/pytests/test_mlModelSaver.py @@ -23,7 +23,7 @@ def test_ensureCLassInstance(): assert mlModelSaverInstance1.baseRelativePath == "test_baseRelativePath" assert mlModelSaverInstance1.modelsFolder == "test_baseRelativePath/test_modelsFolder" tesSupportedModels = mlModelSaverInstance1.showSupportedModels() - assert tesSupportedModels == ['sm.OLS'] + assert tesSupportedModels == ['sm.OLS', 'sm.Logit'] def test_OLS_LinearRegression(): diff --git a/setup.py b/setup.py index 6f67a4b..ccf774c 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ from setuptools import setup, find_packages setup( name='mlModelSaver', - version='1.0.17', + version='1.0.18', 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