postgresml · kczimm · Aug 28, 2024 · Aug 28, 2024 · Aug 28, 2024
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -45,7 +45,7 @@ jobs:
             ~/.cargo
             pgml-extension/target
             ~/.pgrx
-          key: ${{ runner.os }}-rust-1.74-${{ hashFiles('pgml-extension/Cargo.lock') }}-bust2
+          key: ${{ runner.os }}-rust-1.79-${{ hashFiles('pgml-extension/Cargo.lock') }}-bust2
       - name: Install pgrx
         if: steps.pgml_extension_changed.outputs.PGML_EXTENSION_CHANGED_FILES != '0'
         run: |

diff --git a/pgml-extension/Cargo.lock b/pgml-extension/Cargo.lock
diff --git a/pgml-extension/src/api.rs b/pgml-extension/src/api.rs
@@ -188,13 +188,13 @@ fn train_joint(
                 "You must pass a `relation_name` and `y_column_name` to snapshot the first time you train a model.",
             );
 
-            info!("Using existing snapshot from {}", snapshot.snapshot_name(),);
+            notice!("Using existing snapshot from {}", snapshot.snapshot_name(),);
 
             snapshot
         }
 
         Some(relation_name) => {
-            info!(
+            notice!(
                 "Snapshotting table \"{}\", this may take a little while...",
                 relation_name
             );
@@ -213,7 +213,7 @@ fn train_joint(
             );
 
             if materialize_snapshot {
-                info!(
+                notice!(
                     "Snapshot of table \"{}\" created and saved in {}",
                     relation_name,
                     snapshot.snapshot_name(),
@@ -276,9 +276,10 @@ fn train_joint(
                     let deployed_metric = deployed_metrics_obj
                         .get(&default_target_metric)
                         .and_then(|v| v.as_f64());
-                    info!(
+                    notice!(
                         "Comparing to deployed model {}: {:?}",
-                        default_target_metric, deployed_metric
+                        default_target_metric,
+                        deployed_metric
                     );
                     let new_metric = new_metrics.get(&default_target_metric).and_then(|v| v.as_f64());
 
@@ -888,13 +889,13 @@ fn tune(
                 "You must pass a `relation_name` and `y_column_name` to snapshot the first time you train a model.",
             );
 
-            info!("Using existing snapshot from {}", snapshot.snapshot_name(),);
+            notice!("Using existing snapshot from {}", snapshot.snapshot_name(),);
 
             snapshot
         }
 
         Some(relation_name) => {
-            info!(
+            notice!(
                 "Snapshotting table \"{}\", this may take a little while...",
                 relation_name
             );
@@ -909,7 +910,7 @@ fn tune(
             );
 
             if materialize_snapshot {
-                info!(
+                notice!(
                     "Snapshot of table \"{}\" created and saved in {}",
                     relation_name,
                     snapshot.snapshot_name(),
@@ -1619,7 +1620,7 @@ mod tests {
         // to test deployments.
         let setting = Spi::get_one::<String>("select setting from pg_settings where name = 'data_directory'").unwrap();
 
-        info!("Data directory: {}", setting.unwrap());
+        notice!("Data directory: {}", setting.unwrap());
 
         for runtime in [Runtime::python, Runtime::rust] {
             let result: Vec<(String, String, String, bool)> = train(
@@ -1657,7 +1658,7 @@ mod tests {
         // to test deployments.
         let setting = Spi::get_one::<String>("select setting from pg_settings where name = 'data_directory'").unwrap();
 
-        info!("Data directory: {}", setting.unwrap());
+        notice!("Data directory: {}", setting.unwrap());
 
         for runtime in [Runtime::python, Runtime::rust] {
             let result: Vec<(String, String, String, bool)> = train(
@@ -1695,7 +1696,7 @@ mod tests {
         // to test deployments.
         let setting = Spi::get_one::<String>("select setting from pg_settings where name = 'data_directory'").unwrap();
 
-        info!("Data directory: {}", setting.unwrap());
+        notice!("Data directory: {}", setting.unwrap());
 
         for runtime in [Runtime::python, Runtime::rust] {
             let result: Vec<(String, String, String, bool)> = train(

diff --git a/pgml-extension/src/orm/model.rs b/pgml-extension/src/orm/model.rs
@@ -139,7 +139,7 @@ impl Model {
 
         let mut model = model.unwrap();
 
-        info!("Training {}", model);
+        notice!("Training {}", model);
         model.fit(&dataset);
 
         Spi::run_with_args(
@@ -222,7 +222,7 @@ impl Model {
         let id = model.id;
         let path = std::path::PathBuf::from(format!("/tmp/postgresml/models/{id}"));
 
-        info!("Tuning {}", model);
+        notice!("Tuning {}", model);
         let metrics: HashMap<String, f64>;
         match dataset {
             TextDatasetType::TextClassification(dataset) => {
@@ -267,7 +267,7 @@ impl Model {
         };
 
         model.metrics = Some(JsonB(json!(metrics)));
-        info!("Metrics: {:?}", &metrics);
+        notice!("Metrics: {:?}", &metrics);
 
         Spi::get_one_with_args::<i64>(
             "UPDATE pgml.models SET hyperparams = $1, metrics = $2 WHERE id = $3 RETURNING id",
@@ -439,7 +439,7 @@ impl Model {
             }
         }
 
-        info!("Model cache miss {:?}", id);
+        notice!("Model cache miss {:?}", id);
         let model = Arc::new(Model::find(id)?);
         let mut models = DEPLOYED_MODELS_BY_ID.lock();
         models.insert(id, Arc::clone(&model));
@@ -611,7 +611,7 @@ impl Model {
     // The box is borrowed so that it may be reused by the caller
     #[allow(clippy::borrowed_box)]
     fn test(&self, dataset: &Dataset) -> IndexMap<String, f32> {
-        info!("Testing {:?} estimator {:?}", self.project.task, self);
+        notice!("Testing {:?} estimator {:?}", self.project.task, self);
         // Test the estimator on the data
         let y_hat = self.predict_batch(&dataset.x_test).unwrap();
         let y_test = &dataset.y_test;
@@ -724,7 +724,7 @@ impl Model {
         dataset: &Dataset,
         hyperparams: &Hyperparams,
     ) -> (Box<dyn Bindings>, IndexMap<String, f32>) {
-        info!("Hyperparams: {}", serde_json::to_string_pretty(hyperparams).unwrap());
+        notice!("Hyperparams: {}", serde_json::to_string_pretty(hyperparams).unwrap());
 
         let fit = self.get_fit_function();
         let now = Instant::now();
@@ -737,7 +737,7 @@ impl Model {
 
         metrics.insert("fit_time".to_string(), fit_time.as_secs_f32());
         metrics.insert("score_time".to_string(), score_time.as_secs_f32());
-        info!("Metrics: {:?}", &metrics);
+        notice!("Metrics: {:?}", &metrics);
 
         let mut bindings = None;
         std::mem::swap(&mut self.bindings, &mut bindings);
@@ -804,7 +804,7 @@ impl Model {
         let mut all_bindings = Vec::with_capacity(all_hyperparams.len());
         let mut all_metrics = Vec::with_capacity(all_hyperparams.len());
 
-        info!(
+        notice!(
             "Hyperparameter searches: {}, cross validation folds: {}",
             all_hyperparams.len(),
             cv

diff --git a/pgml-extension/src/orm/project.rs b/pgml-extension/src/orm/project.rs
@@ -74,7 +74,7 @@ impl Project {
     }
 
     pub fn deploy(&self, model_id: i64, strategy: Strategy) {
-        info!("Deploying model id: {:?}", model_id);
+        notice!("Deploying model id: {:?}", model_id);
         Spi::get_one_with_args::<i64>(
             "INSERT INTO pgml.deployments (project_id, model_id, strategy) VALUES ($1, $2, $3::pgml.strategy) RETURNING id",
             vec![

diff --git a/pgml-extension/src/orm/snapshot.rs b/pgml-extension/src/orm/snapshot.rs
@@ -334,7 +334,7 @@ impl Column {
             }
         }
 
-        info!("Column {:?}: {:?}", self.name, statistics);
+        notice!("Column {:?}: {:?}", self.name, statistics);
     }
 }
 
@@ -860,7 +860,7 @@ impl Snapshot {
 
         let data = data.unwrap();
 
-        info!("{}", data);
+        notice!("{}", data);
 
         data
     }
@@ -963,7 +963,7 @@ impl Snapshot {
 
         let data = data.unwrap();
 
-        info!("{}", data);
+        notice!("{}", data);
 
         data
     }
@@ -1058,7 +1058,7 @@ impl Snapshot {
 
         let data = data.unwrap();
 
-        info!("{}", data);
+        notice!("{}", data);
 
         data
     }
@@ -1364,7 +1364,7 @@ impl Snapshot {
 
         let data = data.unwrap();
 
-        info!("{}", data);
+        notice!("{}", data);
 
         data
     }