virgo-modules 0.0.67__py3-none-any.whl → 0.0.69__py3-none-any.whl

virgo_modules/src/re_utils.py

@@ -584,7 +584,7 @@ def call_ml_objects(stock_code, client, call_models = False):
     return objects
 
 class produce_plotly_plots:
-    def __init__(self,ticket_name, data_frame,settings, save_path = False, save_aws = False, show_plot= True, aws_credentials = False):
+    def __init__(self,ticket_name, data_frame,settings, save_path = False, save_aws = False, show_plot= True, aws_credentials = False, return_figs = False):
         """
         ticket_name: str asset name
         data_frame: pandas df
@@ -602,6 +602,7 @@ class produce_plotly_plots:
         self.save_aws = save_aws
         self.show_plot = show_plot
         self.aws_credentials = aws_credentials
+        self.return_figs = return_figs
 
     def plot_asset_signals(self, feature_list,spread_column, date_intervals = False):
         
@@ -674,7 +675,9 @@ class produce_plotly_plots:
         if self.save_path and self.save_aws:
             # upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = f'market_plots/{self.ticket_name}/'+result_json_name ,input_path = self.save_path+result_json_name)
             upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = self.save_aws + result_json_name, input_path = self.save_path + result_json_name, aws_credentials = self.aws_credentials)
-
+        if self.return_figs:
+            return fig
+        
     def explore_states_ts(self):
         result_json_name = 'ts_hmm.json'
         df = self.data_frame
@@ -720,8 +723,10 @@ class produce_plotly_plots:
         if self.save_path and self.save_aws:
             # upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = f'market_plots/{self.ticket_name}/'+result_json_name ,input_path = self.save_path+result_json_name)
             upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = self.save_aws + result_json_name, input_path = self.save_path + result_json_name, aws_credentials = self.aws_credentials)
-
-    def plot_hmm_analysis(self,settings, hmm_model, date_intervals = False, model = False):
+        if self.return_figs:
+            return fig
+        
+    def plot_hmm_analysis(self,settings, t_matrix, model = False):
         result_json_name = 'hmm_analysis.json'
         df = self.data_frame
         hmm_n_clust = self.settings['settings']['hmm']['n_clusters']
@@ -744,7 +749,7 @@ class produce_plotly_plots:
 
         ### transition probabilities
         row_i = 1
-        t_matrix = (hmm_model._model_impl['hmm'].transmat_)*100
+        # t_matrix = (hmm_model._model_impl['hmm'].transmat_)*100
         fig.add_trace(go.Heatmap(z = t_matrix, text = np.round(t_matrix,2),texttemplate="%{text}",coloraxis='coloraxis'),row=row_i, col=2)
         fig.update_xaxes(title_text='State To', row=row_i, col=2)
         fig.update_yaxes(title_text='State From', row=row_i, col=2)
@@ -839,7 +844,9 @@ class produce_plotly_plots:
             
             upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = self.save_aws + result_json_name, input_path = self.save_path + result_json_name, aws_credentials = self.aws_credentials)
             upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = self.save_aws + 'market_message.json', input_path = self.save_path + 'market_message.json', aws_credentials = self.aws_credentials)
-
+        
+        if self.return_figs:
+            return fig, messages
     def produce_forecasting_plot(self,predictions):
         result_json_name = 'forecast_plot.json'
         hmm_n_clust = self.settings['settings']['hmm']['n_clusters']
@@ -1034,9 +1041,9 @@ def produce_simple_ts_from_model(stock_code, configs, n_days = 2000 , window_sco
                 
     fig.update_layout(height=height_plot, width=1600, title_text = f'asset plot and signals: {stock_code}')
 
-    del object_stock, df
+    del object_stock
     
-    return fig
+    return fig, df
 
 def save_edge_model(data, save_path = False, save_aws = False, show_result = False, aws_credentials = False):
     """

{virgo_modules-0.0.67.dist-info → virgo_modules-0.0.69.dist-info}/METADATA

@@ -1,53 +1,53 @@
-Metadata-Version: 2.1
-Name: virgo-modules
-Version: 0.0.67
-Summary: data processing and statistical modeling using stock market data
-Home-page: https://github.com/miguelmayhem92/virgo_module
-Author: Miguel Mayhuire
-Author-email: miguelmayhem92@gmail.com
-License: MIT
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.9, <3.10
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: feature-engine (==1.6.1)
-Requires-Dist: matplotlib (==3.6.3)
-Requires-Dist: mlflow (==2.1.1)
-Requires-Dist: numpy (==1.23.5)
-Requires-Dist: optuna (==3.1.0)
-Requires-Dist: pandas (==1.5.3)
-Requires-Dist: plotly (==5.15.0)
-Requires-Dist: rsa (==4.9)
-Requires-Dist: scikit-learn (==1.2.1)
-Requires-Dist: scipy (==1.10.0)
-Requires-Dist: seaborn (==0.12.2)
-Requires-Dist: starlette (==0.22.0)
-Requires-Dist: statsmodels (==0.13.5)
-Requires-Dist: ta (==0.10.2)
-Requires-Dist: yfinance (==0.2.9)
-Requires-Dist: hmmlearn (==0.3.0)
-Requires-Dist: boto3
-Provides-Extra: dev
-Requires-Dist: pytest (>=7.0) ; extra == 'dev'
-
-# Virgo Package
-
-this package contains the utils and helper functions that used in virgo project
-
-### how to use
-
-istall using 
-
-```
-pip install virgo-modules
-```
-
-geting data:
-
-```
-obj = stock_eda_panel(stock_code = 'PEP', n_days = 20)
-obj.get_data()
-print(obj.df.shape)
-```
+Metadata-Version: 2.1
+Name: virgo-modules
+Version: 0.0.69
+Summary: data processing and statistical modeling using stock market data
+Home-page: https://github.com/miguelmayhem92/virgo_module
+Author: Miguel Mayhuire
+Author-email: miguelmayhem92@gmail.com
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.9, <3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: feature-engine ==1.6.1
+Requires-Dist: matplotlib ==3.6.3
+Requires-Dist: mlflow ==2.1.1
+Requires-Dist: numpy ==1.23.5
+Requires-Dist: optuna ==3.1.0
+Requires-Dist: pandas ==1.5.3
+Requires-Dist: plotly ==5.15.0
+Requires-Dist: rsa ==4.9
+Requires-Dist: scikit-learn ==1.2.1
+Requires-Dist: scipy ==1.10.0
+Requires-Dist: seaborn ==0.12.2
+Requires-Dist: starlette ==0.22.0
+Requires-Dist: statsmodels ==0.13.5
+Requires-Dist: ta ==0.10.2
+Requires-Dist: yfinance ==0.2.9
+Requires-Dist: hmmlearn ==0.3.0
+Requires-Dist: boto3
+Provides-Extra: dev
+Requires-Dist: pytest >=7.0 ; extra == 'dev'
+
+# Virgo Package
+
+this package contains the utils and helper functions that used in virgo project
+
+### how to use
+
+istall using 
+
+```
+pip install virgo-modules
+```
+
+geting data:
+
+```
+obj = stock_eda_panel(stock_code = 'PEP', n_days = 20)
+obj.get_data()
+print(obj.df.shape)
+```

{virgo_modules-0.0.67.dist-info → virgo_modules-0.0.69.dist-info}/RECORD

@@ -3,10 +3,10 @@ virgo_modules/src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 virgo_modules/src/aws_utils.py,sha256=GWmVdXM0mIJJPn-X-bEtM4KtNPCHM1D457hnuKxaM7E,1383
 virgo_modules/src/edge_utils.py,sha256=Ihdmq7dyb8gOvG6CrDal7wsa15tqsdsFk6KINwM6578,7691
 virgo_modules/src/pull_artifacts.py,sha256=5OPrgR7pcMSdpbevDRhf0ebk7g7ZRjff4NpTIIWAKjE,1989
-virgo_modules/src/re_utils.py,sha256=s6fbjOI59Clw5f1sVfqexiviYJyRVGobYe7qFsC-tQo,52211
+virgo_modules/src/re_utils.py,sha256=lS7Zx0PGT6aWRcL5kAAfGMt1Q0J1y2IV2fnnCqma-io,52445
 virgo_modules/src/ticketer_source.py,sha256=_vJkF38yAl2FbxvcOWHbjG5n7o34lh8SdRR8zzIxChY,101408
-virgo_modules-0.0.67.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.0.67.dist-info/METADATA,sha256=Y5PmJf9G5ZC04DYJcfP1lZJl6IL7oKU8MxIWqjc5FHg,1410
-virgo_modules-0.0.67.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-virgo_modules-0.0.67.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.0.67.dist-info/RECORD,,
+virgo_modules-0.0.69.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.0.69.dist-info/METADATA,sha256=RVCh9ZFZ3G9Z1mklQJwOatQSSzOFgBMVpF0jFpg-gaY,1429
+virgo_modules-0.0.69.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+virgo_modules-0.0.69.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.0.69.dist-info/RECORD,,

{virgo_modules-0.0.67.dist-info → virgo_modules-0.0.69.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.37.1)
+Generator: bdist_wheel (0.41.2)
 Root-Is-Purelib: true
 Tag: py3-none-any