sovai 0.2.58__tar.gz → 0.2.59__tar.gz

{sovai-0.2.58 → sovai-0.2.59}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sovai
-Version: 0.2.58
+Version: 0.2.59
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow

{sovai-0.2.58 → sovai-0.2.59}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "sovai"
-version = "0.2.58"
+version = "0.2.59"
 description = "python for asset management"
 authors = [
     {name = "Derek Snow", email = "d.snow@outlook.com"}

{sovai-0.2.58 → sovai-0.2.59}/sovai/extensions/pandas_extensions.py

@@ -8,9 +8,8 @@ from numpy.linalg import norm
 # from sovai.extensions.pfa_feature_selector import run_pfa_simulations
 pd.options.display.float_format = "{:.3f}".format
 
-from sovai.extensions.shapley_global_importance import run_simulations_global_importance, run_simulations_frame_global
 
-from sovai.extensions.shapley_importance import run_simulations_frame
+# from sovai.extensions.shapley_importance import run_simulations_frame
 
 from sovai.extensions.feature_neutralizer import (
     neutralize_features_function,
@@ -758,6 +757,8 @@ class CustomDataFrame(pd.DataFrame):
 
     def importance(self, method="random_projection"):
 
+        from sovai.extensions.shapley_global_importance import run_simulations_global_importance
+
         if method == "random_projection":
             return random_projection_importance(self)
         elif method == "fourier":
@@ -1013,6 +1014,8 @@ class CustomDataFrame(pd.DataFrame):
         :param clustering_method: The clustering method to use ('OPTICS' or 'KMeans').
         :return: A DataFrame with average SHAP values per feature.
         """
+        from sovai.extensions.shapley_global_importance import  run_simulations_frame_global
+
         # Running the simulations and getting average SHAP values
         avg_shap_values = run_simulations_frame_global(
             self, num_simulations=num_simulations, clustering_method=clustering_method