sovai 0.2.23__tar.gz → 0.2.24__tar.gz

{sovai-0.2.23 → sovai-0.2.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sovai
-Version: 0.2.23
+Version: 0.2.24
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow
@@ -15,6 +15,7 @@ Provides-Extra: full
 Requires-Dist: boto3 (>=1.20)
 Requires-Dist: dash (>=2.0) ; extra == "full"
 Requires-Dist: dash-bootstrap-components (>=1.6.0) ; extra == "full"
+Requires-Dist: edgar-tool (>=1.3.3,<2.0.0)
 Requires-Dist: great-tables (>=0.9) ; extra == "full"
 Requires-Dist: ipywidgets (>=8.1.3) ; extra == "full"
 Requires-Dist: joblib (>=1.0)

{sovai-0.2.23 → sovai-0.2.24}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sovai"
-version = "0.2.23"
+version = "0.2.24"
 description = "python for asset management"
 authors = ["Derek Snow <d.snow@outlook.com>"]
 license = "Proprietary License"
@@ -60,6 +60,7 @@ 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"
+edgar-tool = "^1.3.3"
 
 [tool.poetry.extras]
 full = [

{sovai-0.2.23 → sovai-0.2.24}/sovai/__init__.py

@@ -7,7 +7,7 @@ from .basic_auth import basic_auth
 from .token_auth import token_auth
 from .utils.file_management import save_or_update_tickers, save_or_update_codes
 import warnings
-warnings.filterwarnings("ignore", category=FutureWarning, module="dask.dataframe")
+warnings.filterwarnings("ignore", category=FutureWarning)
 
 
 __version__ = "0.1.25"

{sovai-0.2.23 → sovai-0.2.24}/sovai/plots/bankruptcy/bankruptcy_plots.py

@@ -124,9 +124,10 @@ def plot_bankruptcy_monthly_line(df, tickers=None):
     # Set default tickers if None is provided
     if tickers is None:
         tickers = ["TSLA", "LULU"]
-        
+    
     # Convert the tickers list to a string (limit to 5 tickers)
     tickers_str = ", ".join(tickers[:5])
+    
     ticker_input = widgets.Text(value=tickers_str, description="Tickers:")
     probability_column_selector = widgets.Dropdown(
         options=df.columns, value="probability", description="Probability"
@@ -140,15 +141,13 @@ def plot_bankruptcy_monthly_line(df, tickers=None):
     same_plot_selector = widgets.Checkbox(value=False, description="Same Plot")
     dark_mode_selector = widgets.Checkbox(value=True, description="Dark Mode")
 
-    # Update the plot_button styling
-    plot_button = widgets.Button(
-        description="Plot", layout=widgets.Layout(border="solid 1px blue")
-    )
+    # Create an output widget to display the plot
+    output = widgets.Output()
 
     # Create a function to update the plot
     def update_plot(change):
         with output:
-            clear_output(wait=True)
+            output.clear_output(wait=True)
             try:
                 tickers = [ticker.strip() for ticker in ticker_input.value.split(",")]
                 plot_ticker_probabilities(
@@ -175,18 +174,8 @@ def plot_bankruptcy_monthly_line(df, tickers=None):
     same_plot_selector.observe(update_plot, names="value")
     dark_mode_selector.observe(update_plot, names="value")
 
-    # Remove the on_click event from plot_button as it's no longer needed
-    # plot_button.on_click(on_plot_button_clicked)
-
-    # Create an output widget to display the plot
-    output = widgets.Output()
-
-    # Initially, display the current plot
-    update_plot(None)  # Pass None because the event parameter is not used
-
     # Function to toggle the visibility of the widget controls
     def toggle_controls(b):
-        # Modify to collapse the controls
         controls_container.layout.display = (
             "none" if controls_container.layout.display == "flex" else "flex"
         )
@@ -207,6 +196,10 @@ def plot_bankruptcy_monthly_line(df, tickers=None):
     # Display the widgets and the output widget
     display(controls_container, output)
 
+    # Initially, display the current plot
+    update_plot(None)  # Pass None because the event parameter is not used
+
+
 
 def plot_pca_clusters(df, target="target", max_lag=12, max_date=None):
     """