virgo-modules 0.0.65__py3-none-any.whl → 0.0.66__py3-none-any.whl

virgo_modules/src/ticketer_source.py

@@ -197,7 +197,7 @@ class stock_eda_panel(object):
 
         stock = yf.Ticker(self.stock_code)
         df = stock.history(period=self.data_window)
-        
+
         df = df.sort_values('Date')
         df.reset_index(inplace=True)
         df['Date'] = pd.to_datetime(df['Date'], format='mixed',utc=True).dt.date
@@ -1687,7 +1687,7 @@ class hmm_feature_selector():
         
 class signal_analyser_object:
     
-    def __init__(self, data,symbol_name, show_plot = True, save_path = False, save_aws = False, aws_credentials = False):
+    def __init__(self, data,symbol_name, show_plot = True, save_path = False, save_aws = False, aws_credentials = False, return_fig = False):
         """
         data: pandas df
         symbol_name: str name of the asset
@@ -1695,6 +1695,7 @@ class signal_analyser_object:
         save_path: str local path for saving e.g r'C:/path/to/the/file/'
         save_aws: str remote key in s3 bucket path e.g. 'path/to/file/'
         aws_credentials: dict
+        return_fig: boolean return the image function as result
         """
         self.data = data.copy()
         self.ticket_name = symbol_name
@@ -1702,6 +1703,7 @@ class signal_analyser_object:
         self.save_path = save_path
         self.save_aws = save_aws
         self.aws_credentials = aws_credentials
+        self.return_fig = return_fig
 
     def signal_analyser(self, test_size, feature_name, days_list, threshold = 0.05,verbose = False, signal_position = False):
         data = self.data
@@ -1818,7 +1820,14 @@ class signal_analyser_object:
             upload_file_to_aws(bucket = 'VIRGO_BUCKET', key = self.save_aws + result_plot_name, input_path = self.save_path + result_plot_name, aws_credentials = self.aws_credentials)
         if not self.show_plot:
             plt.close()
+
         del df
+
+        if self.return_fig:
+            return fig
+        
+
+        
         
     def create_backtest_signal(self,days_strategy, test_size, feature_name):
         asset_1 = 'Close'
@@ -1918,6 +1927,9 @@ class signal_analyser_object:
             plt.close()
             
         del df1,df2,dft
+
+        if self.return_fig:
+            return fig
         
 def execute_signal_analyser(test_data_size, feature_name, days_list, configuration, method, object_stock, signal_analyser_object, plot = False, backtest= False):
     method(**configuration)

{virgo_modules-0.0.65.dist-info → virgo_modules-0.0.66.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: virgo-modules
-Version: 0.0.65
+Version: 0.0.66
 Summary: data processing and statistical modeling using stock market data
 Home-page: https://github.com/miguelmayhem92/virgo_module
 Author: Miguel Mayhuire

{virgo_modules-0.0.65.dist-info → virgo_modules-0.0.66.dist-info}/RECORD

@@ -4,9 +4,9 @@ virgo_modules/src/aws_utils.py,sha256=GWmVdXM0mIJJPn-X-bEtM4KtNPCHM1D457hnuKxaM7
 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/ticketer_source.py,sha256=6jXRHxZOwu2AekFjdKMeSdtamZ01FWYaFjfPgklw4rE,101171
-virgo_modules-0.0.65.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
-virgo_modules-0.0.65.dist-info/METADATA,sha256=zLZiojc9w7hP0EMo4PxaiMFWZYCo8LTgLOuU_0oF_44,1429
-virgo_modules-0.0.65.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-virgo_modules-0.0.65.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
-virgo_modules-0.0.65.dist-info/RECORD,,
+virgo_modules/src/ticketer_source.py,sha256=Jn0cu6pkMtUG2tIhBvdgocxIJDt2II4TP0Yc85rgiQM,101420
+virgo_modules-0.0.66.dist-info/LICENSE,sha256=pNgFyCYgmimaw0o6V20JupZLROycAnOA_HDDh1tX2V4,1097
+virgo_modules-0.0.66.dist-info/METADATA,sha256=n33RkrFJXxii_Ro0tOM4tOTHkLRbnpK3FUwIaUMPc3k,1429
+virgo_modules-0.0.66.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+virgo_modules-0.0.66.dist-info/top_level.txt,sha256=ZjI-qEkDtT-8mFwGAWnXfqPOKEGlIhWRW1es1VyXc60,14
+virgo_modules-0.0.66.dist-info/RECORD,,