upgrade to latest rc logging

ahosler · ahosler · commit 12e24d54a1df · 2024-11-12T16:30:51.000Z
diff --git a/ads/opctl/operator/lowcode/anomaly/model/anomaly_merlion.py b/ads/opctl/operator/lowcode/anomaly/model/anomaly_merlion.py
@@ -23,7 +23,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class AnomalyMerlionOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/automlx.py b/ads/opctl/operator/lowcode/anomaly/model/automlx.py
@@ -15,7 +15,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class AutoMLXOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/autots.py b/ads/opctl/operator/lowcode/anomaly/model/autots.py
@@ -15,7 +15,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class AutoTSOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/base_model.py b/ads/opctl/operator/lowcode/anomaly/model/base_model.py
@@ -35,7 +35,7 @@
 from ..operator_config import AnomalyOperatorConfig, AnomalyOperatorSpec
 from .anomaly_dataset import AnomalyDatasets, AnomalyOutput, TestData
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class AnomalyOperatorBaseModel(ABC):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/isolationforest.py b/ads/opctl/operator/lowcode/anomaly/model/isolationforest.py
@@ -15,7 +15,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class IsolationForestOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/oneclasssvm.py b/ads/opctl/operator/lowcode/anomaly/model/oneclasssvm.py
@@ -15,7 +15,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class OneClassSVMOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py b/ads/opctl/operator/lowcode/anomaly/model/randomcutforest.py
@@ -16,7 +16,7 @@
 from .anomaly_dataset import AnomalyOutput
 from .base_model import AnomalyOperatorBaseModel
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class RandomCutForestOperatorModel(AnomalyOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/forecast/model/arima.py b/ads/opctl/operator/lowcode/forecast/model/arima.py
@@ -20,7 +20,7 @@
 from .base_model import ForecastOperatorBaseModel
 from .forecast_datasets import ForecastDatasets, ForecastOutput
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class ArimaOperatorModel(ForecastOperatorBaseModel):
diff --git a/ads/opctl/operator/lowcode/forecast/model/automlx.py b/ads/opctl/operator/lowcode/forecast/model/automlx.py
@@ -24,7 +24,7 @@
 from .base_model import ForecastOperatorBaseModel
 from .forecast_datasets import ForecastDatasets, ForecastOutput
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 AUTOMLX_N_ALGOS_TUNED = 4
 AUTOMLX_DEFAULT_SCORE_METRIC = "neg_sym_mean_abs_percent_error"
 
@@ -78,10 +78,10 @@ def preprocess(self, data):  # TODO: re-use self.le for explanations
     def _build_model(self) -> pd.DataFrame:
         import logging
 
-        import automlx
+        from automlx import Pipeline, init
 
         try:
-            automlx.init(
+            init(
                 engine="ray",
                 engine_opts={"ray_setup": {"_temp_dir": "/tmp/ray-temp"}},
                 loglevel=logging.CRITICAL,
@@ -123,7 +123,7 @@ def _build_model(self) -> pd.DataFrame:
                 if self.loaded_models is not None and s_id in self.loaded_models:
                     model = self.loaded_models[s_id]
                 else:
-                    model = automlx.Pipeline(
+                    model = Pipeline(
                         task="forecasting",
                         **model_kwargs,
                     )
diff --git a/ads/opctl/operator/lowcode/forecast/model/autots.py b/ads/opctl/operator/lowcode/forecast/model/autots.py
@@ -20,7 +20,7 @@
 from .base_model import ForecastOperatorBaseModel
 from .forecast_datasets import ForecastDatasets, ForecastOutput
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 AUTOTS_MAX_GENERATION = 10
 AUTOTS_MODELS_TO_VALIDATE = 0.15
 
diff --git a/ads/opctl/operator/lowcode/forecast/model/base_model.py b/ads/opctl/operator/lowcode/forecast/model/base_model.py
@@ -51,7 +51,7 @@
 from ..operator_config import ForecastOperatorConfig, ForecastOperatorSpec
 from .forecast_datasets import ForecastDatasets
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class ForecastOperatorBaseModel(ABC):
diff --git a/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py b/ads/opctl/operator/lowcode/forecast/model/ml_forecast.py
@@ -193,7 +193,7 @@ def _generate_report(self):
         import report_creator as rc
         from utilsforecast.plotting import plot_series
 
-        logging.getLogger("root").setLevel(logging.WARNING)
+        logging.getLogger("report_creator").setLevel(logging.WARNING)
 
         # Section 1: Forecast Overview
         sec1_text = rc.Block(
diff --git a/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py b/ads/opctl/operator/lowcode/forecast/model/neuralprophet.py
@@ -308,7 +308,7 @@ def objective(trial):
     def _generate_report(self):
         import report_creator as rc
 
-        logging.getLogger("root").setLevel(logging.WARNING)
+        logging.getLogger("report_creator").setLevel(logging.WARNING)
 
         series_ids = self.models.keys()
         all_sections = []
diff --git a/ads/opctl/operator/lowcode/forecast/model/prophet.py b/ads/opctl/operator/lowcode/forecast/model/prophet.py
@@ -243,7 +243,7 @@ def _generate_report(self):
         import report_creator as rc
         from prophet.plot import add_changepoints_to_plot
 
-        logging.getLogger("root").setLevel(logging.WARNING)
+        logging.getLogger("report_creator").setLevel(logging.WARNING)
 
         series_ids = self.models.keys()
         all_sections = []
diff --git a/ads/opctl/operator/lowcode/forecast/utils.py b/ads/opctl/operator/lowcode/forecast/utils.py
@@ -32,7 +32,7 @@
 
 from .const import RENDER_LIMIT, SupportedMetrics
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 def _label_encode_dataframe(df, no_encode: Set = None):
diff --git a/ads/opctl/operator/lowcode/pii/model/report.py b/ads/opctl/operator/lowcode/pii/model/report.py
@@ -46,7 +46,7 @@
         f"`pip install {OptionalDependency.PII}`."
     ) from e
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 @dataclass(repr=True)
diff --git a/ads/opctl/operator/lowcode/recommender/model/base_model.py b/ads/opctl/operator/lowcode/recommender/model/base_model.py
@@ -29,7 +29,7 @@
 from .factory import SupportedModels
 from .recommender_dataset import RecommenderDatasets
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class RecommenderOperatorBaseModel(ABC):
diff --git a/ads/opctl/operator/lowcode/recommender/model/svd.py b/ads/opctl/operator/lowcode/recommender/model/svd.py
@@ -16,7 +16,7 @@
 from .factory import RecommenderOperatorBaseModel
 from .recommender_dataset import RecommenderDatasets
 
-logging.getLogger("root").setLevel(logging.WARNING)
+logging.getLogger("report_creator").setLevel(logging.WARNING)
 
 
 class SVDOperatorModel(RecommenderOperatorBaseModel):
diff --git a/pyproject.toml b/pyproject.toml
@@ -171,13 +171,13 @@ forecast = [
   "statsmodels",
   "plotly",
   "oracledb",
-  "report-creator==1.0.27",
+  "report-creator==1.0.28",
 ]
 anomaly  = [
   "oracle_ads[opctl]",
   "autots",
   "oracledb",
-  "report-creator==1.0.27",
+  "report-creator==1.0.28",
   "rrcf==0.4.4",
   "scikit-learn",
   "salesforce-merlion[all]==2.0.4"
@@ -186,7 +186,7 @@ recommender = [
   "oracle_ads[opctl]",
   "scikit-surprise",
   "plotly",
-  "report-creator==1.0.27",
+  "report-creator==1.0.28",
 ]
 feature-store-marketplace = [
     "oracle-ads[opctl]",
@@ -202,7 +202,7 @@ pii = [
   "scrubadub_spacy",
   "spacy-transformers==1.2.5",
   "spacy==3.6.1",
-  "report-creator==1.0.27",
+  "report-creator==1.0.28",
 ]
 llm = ["langchain>=0.2", "langchain-community", "langchain_openai", "pydantic>=2,<3", "evaluate>=0.4.0"]
 aqua = ["jupyter_server"]
