virgo-modules 0.4.2__tar.gz → 0.4.3__tar.gz

{virgo_modules-0.4.2 → virgo_modules-0.4.3}/PKG-INFO

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

{virgo_modules-0.4.2 → virgo_modules-0.4.3}/setup.py

@@ -5,7 +5,7 @@ with open("virgo_app/README.md", "r") as f:
 
 setup(
     name="virgo_modules",
-    version="0.4.2",
+    version="0.4.3",
     description="data processing and statistical modeling using stock market data",
     package_dir={"": "virgo_app"},
     packages=find_packages(where="virgo_app"),

{virgo_modules-0.4.2 → virgo_modules-0.4.3}/virgo_app/virgo_modules/src/re_utils.py

@@ -792,7 +792,7 @@ def apply_KF(self, trends):
         
 stock_eda_panel.apply_KF = apply_KF
 
-def call_ml_objects(stock_code, client, call_models = False):
+def call_ml_objects(stock_code, client, call_models = False, clean_name=False):
     '''
     call artifcats from mlflow
 
@@ -805,7 +805,11 @@ def call_ml_objects(stock_code, client, call_models = False):
     '''
     objects = dict()
     
-    registered_model_name = f'{stock_code}_models'
+    if clean_name:
+        renamed_stock_code = stock_code.replace("^","__",).replace(".","__").replace("=","__").replace("-","__")
+        registered_model_name = f'{renamed_stock_code}_models'
+    else:
+        registered_model_name = f'{stock_code}_models'
     latest_version_info = client.get_latest_versions(registered_model_name, stages=["Production"])
     latest_production_version = latest_version_info[0].version
     run_id_prod_model = latest_version_info[0].run_id
@@ -816,18 +820,27 @@ def call_ml_objects(stock_code, client, call_models = False):
     )
 
      ## calling models
-    
+    if clean_name:
+        path_hmm = f"runs:/{run_id_prod_model}/{renamed_stock_code}-hmm-model"
+    else:
+        path_hmm = f"runs:/{run_id_prod_model}/{stock_code}-hmm-model"
+
     hmm_model = mlflow.pyfunc.load_model(
-            f"runs:/{run_id_prod_model}/{stock_code}-hmm-model",
-             suppress_warnings = True
+            path_hmm,
+            suppress_warnings = True
             )
     objects['called_hmm_models'] = hmm_model
     
     if call_models:
         
+        if clean_name:
+            path_model = f"runs:/{run_id_prod_model}/{renamed_stock_code}-forecasting-model"
+        else:
+            path_model = f"runs:/{run_id_prod_model}/{stock_code}-forecasting-model"
+
         forecasting_model = mlflow.pyfunc.load_model(
-            f"runs:/{run_id_prod_model}/{stock_code}-forecasting-model",
-             suppress_warnings = True
+            path_model,
+            suppress_warnings = True
             )
         objects['called_forecasting_model'] = forecasting_model
         

{virgo_modules-0.4.2 → virgo_modules-0.4.3}/virgo_app/virgo_modules.egg-info/PKG-INFO

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