sovai 0.2.20__tar.gz → 0.2.22__tar.gz

{sovai-0.2.20 → sovai-0.2.22}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sovai
-Version: 0.2.20
+Version: 0.2.22
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow
@@ -32,6 +32,7 @@ Requires-Dist: polars-talib (==0.1.3) ; extra == "full"
 Requires-Dist: pyarrow (>=5.0)
 Requires-Dist: python-dateutil (>=2.8)
 Requires-Dist: python-dotenv (>=0.10)
+Requires-Dist: rapidfuzz (>=3.9.7,<4.0.0)
 Requires-Dist: requests (>=2.20)
 Requires-Dist: ruptures (>=1.0) ; extra == "full"
 Requires-Dist: scikit-learn (>=1.0) ; extra == "full"

{sovai-0.2.20 → sovai-0.2.22}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sovai"
-version = "0.2.20"
+version = "0.2.22"
 description = "python for asset management"
 authors = ["Derek Snow <d.snow@outlook.com>"]
 license = "Proprietary License"
@@ -59,6 +59,7 @@ lightgbm = {version = ">=4.5.0", optional = true}
 ipywidgets = {version = ">=8.1.3", optional = true}
 polars-talib = {version = "0.1.3", optional = true}
 dash-bootstrap-components = {version = ">=1.6.0", optional = true}
+rapidfuzz = "^3.9.7"
 
 [tool.poetry.extras]
 full = [

{sovai-0.2.20 → sovai-0.2.22}/sovai/get_data.py

@@ -22,6 +22,7 @@ from sovai.utils.stream import stream_data, stream_data_pyarrow
 from sovai.utils.datetime_formats import datetime_format
 from sovai.utils.client_side import client_side_frame
 
+## Note run these for TEST - IT COULD BE HERE some silent problems
 try:
     from sovai.extensions.pandas_extensions import CustomDataFrame
     from sovai.utils.plot import plotting_data
@@ -620,7 +621,7 @@ def data(
                 verbose_mode.vprint(f"header: not pyarrow")
                 data = stream_data(res)
             
-            if HAS_FULL_INSTALL and not use_polars and not isinstance(data, CustomDataFrame):
+            if HAS_FULL_INSTALL:
                 data = CustomDataFrame(data)
             else:
                 data = pd.DataFrame(data)