Bug fix. Remove wrong imports in base model class (#67)

* Bug fix. Remove wrong imports in base model class

* readme update for pypi

.github/workflows/release.yml

@@ -14,7 +14,6 @@ jobs:
     - name: Build package
       run: pip install wheel && python setup.py sdist bdist_wheel && ls -l dist
     - name: Publish package to TestPyPI
-      if: "github.event.release.prerelease"
       uses: pypa/gh-action-pypi-publish@master
       with:
         user: __token__

README.md

@@ -100,7 +100,7 @@ TileDB-ML can be installed:
 
       pip install git+https://github.com/TileDB-Inc/TileDB-ML.git@master
 
-- Soon from PyPi:
+- from PyPi:
 
 [comment]: <> (TileDB-ML is available from either [PyPI]&#40;https://test.pypi.org/project/tiledb-ml/0.1.2.2/&#41; with ``pip``:)
 

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = tiledb-ml
-version = 0.2.1
+version = 0.2.2
 description = Package supports all machine learning functionality for TileDB Embedded and TileDB Cloud
 author = TileDB, Inc.
 author_email = help@tiledb.io

tiledb/ml/models/base.py

@@ -6,13 +6,9 @@
 import json
 import platform
 
-from typing import Optional, Union
+from typing import Optional
 from enum import Enum, unique
 
-from torch.nn import Module
-from tensorflow.python.keras.models import Model
-from sklearn.base import BaseEstimator
-
 
 @unique
 class ModelFileProperties(Enum):
@@ -41,7 +37,7 @@ def __init__(
         uri: str,
         namespace: str = None,
         ctx: tiledb.Ctx = None,
-        model: Union[Module, Model, BaseEstimator] = None,
+        model=None,
     ):
         """
         Base class for saving machine learning models as TileDB arrays