snowflake-ml-python 1.11.0__py3-none-any.whl → 1.12.0__py3-none-any.whl

snowflake/ml/modeling/neighbors/k_neighbors_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/k_neighbors_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/kernel_density.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/local_outlier_factor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/nearest_centroid.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/nearest_neighbors.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/neighborhood_components_analysis.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/radius_neighbors_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neighbors/radius_neighbors_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/bernoulli_rbm.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/mlp_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/neural_network/mlp_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/preprocessing/polynomial_features.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/semi_supervised/label_propagation.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/semi_supervised/label_spreading.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/linear_svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/linear_svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/nu_svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/nu_svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/svc.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/svm/svr.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/decision_tree_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/decision_tree_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/extra_tree_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/tree/extra_tree_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgb_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgb_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgbrf_classifier.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/modeling/xgboost/xgbrf_regressor.py

@@ -60,7 +60,7 @@ DATAFRAME_TYPE = Union[DataFrame, pd.DataFrame]
 
 INFER_SIGNATURE_MAX_ROWS = 100
 
-SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.6')
+SKLEARN_LOWER, SKLEARN_UPPER = ('1.4', '1.7')
 # Modeling library estimators require a smaller sklearn version range.
 if not version.Version(SKLEARN_LOWER) <= version.Version(sklearn.__version__) < version.Version(SKLEARN_UPPER):
     raise Exception(

snowflake/ml/monitoring/_client/model_monitor_sql_client.py

@@ -1,3 +1,4 @@
+from enum import Enum, auto
 from typing import Any, Mapping, Optional
 
 from snowflake import snowpark
@@ -15,6 +16,25 @@ MODEL_JSON_MODEL_NAME_FIELD = "model_name"
 MODEL_JSON_VERSION_NAME_FIELD = "version_name"
 
 
+class MonitorOperation(Enum):
+    SUSPEND = auto()
+    RESUME = auto()
+    ADD = auto()
+    DROP = auto()
+
+    @property
+    def supported_target_properties(self) -> frozenset[str]:
+        return _OPERATION_SUPPORTED_PROPS[self]
+
+
+_OPERATION_SUPPORTED_PROPS: dict[MonitorOperation, frozenset[str]] = {
+    MonitorOperation.SUSPEND: frozenset(),
+    MonitorOperation.RESUME: frozenset(),
+    MonitorOperation.ADD: frozenset({"SEGMENT_COLUMN"}),
+    MonitorOperation.DROP: frozenset({"SEGMENT_COLUMN"}),
+}
+
+
 def _build_sql_list_from_columns(columns: list[sql_identifier.SqlIdentifier]) -> str:
     sql_list = ", ".join([f"'{column}'" for column in columns])
     return f"({sql_list})"
@@ -70,11 +90,17 @@ class ModelMonitorSQLClient:
         baseline_database: Optional[sql_identifier.SqlIdentifier] = None,
         baseline_schema: Optional[sql_identifier.SqlIdentifier] = None,
         baseline: Optional[sql_identifier.SqlIdentifier] = None,
+        segment_columns: Optional[list[sql_identifier.SqlIdentifier]] = None,
         statement_params: Optional[dict[str, Any]] = None,
     ) -> None:
         baseline_sql = ""
         if baseline:
             baseline_sql = f"""BASELINE={self._infer_qualified_schema(baseline_database, baseline_schema)}.{baseline}"""
+
+        segment_columns_sql = ""
+        if segment_columns:
+            segment_columns_sql = f"SEGMENT_COLUMNS={_build_sql_list_from_columns(segment_columns)}"
+
         query_result_checker.SqlResultValidator(
             self._sql_client._session,
             f"""
@@ -93,6 +119,7 @@ class ModelMonitorSQLClient:
                     TIMESTAMP_COLUMN='{timestamp_column}'
                     REFRESH_INTERVAL='{refresh_interval}'
                     AGGREGATION_WINDOW='{aggregation_window}'
+                    {segment_columns_sql}
                     {baseline_sql}""",
             statement_params=statement_params,
         ).has_column("status").has_dimensions(1, 1).validate()
@@ -182,6 +209,7 @@ class ModelMonitorSQLClient:
         actual_score_columns: list[sql_identifier.SqlIdentifier],
         actual_class_columns: list[sql_identifier.SqlIdentifier],
         id_columns: list[sql_identifier.SqlIdentifier],
+        segment_columns: Optional[list[sql_identifier.SqlIdentifier]] = None,
     ) -> None:
         """Ensures all columns exist in the source table.
 
@@ -193,11 +221,14 @@ class ModelMonitorSQLClient:
             actual_score_columns: List of actual score column names.
             actual_class_columns: List of actual class column names.
             id_columns: List of id column names.
+            segment_columns: List of segment column names.
 
         Raises:
             ValueError: If any of the columns do not exist in the source.
         """
 
+        segment_columns = [] if segment_columns is None else segment_columns
+
         if timestamp_column not in source_column_schema:
             raise ValueError(f"Timestamp column {timestamp_column} does not exist in source.")
 
@@ -214,6 +245,9 @@ class ModelMonitorSQLClient:
         if not all([column_name in source_column_schema for column_name in id_columns]):
             raise ValueError(f"ID column(s): {id_columns} do not exist in source.")
 
+        if not all([column_name in source_column_schema for column_name in segment_columns]):
+            raise ValueError(f"Segment column(s): {segment_columns} do not exist in source.")
+
     def validate_source(
         self,
         *,
@@ -226,7 +260,9 @@ class ModelMonitorSQLClient:
         actual_score_columns: list[sql_identifier.SqlIdentifier],
         actual_class_columns: list[sql_identifier.SqlIdentifier],
         id_columns: list[sql_identifier.SqlIdentifier],
+        segment_columns: Optional[list[sql_identifier.SqlIdentifier]] = None,
     ) -> None:
+
         source_database = source_database or self._database_name
         source_schema = source_schema or self._schema_name
         # Get Schema of the source. Implicitly validates that the source exists.
@@ -244,19 +280,38 @@ class ModelMonitorSQLClient:
             actual_score_columns=actual_score_columns,
             actual_class_columns=actual_class_columns,
             id_columns=id_columns,
+            segment_columns=segment_columns,
         )
 
     def _alter_monitor(
         self,
-        operation: str,
+        operation: MonitorOperation,
         monitor_name: sql_identifier.SqlIdentifier,
+        target_property: Optional[str] = None,
+        target_value: Optional[sql_identifier.SqlIdentifier] = None,
         statement_params: Optional[dict[str, Any]] = None,
     ) -> None:
-        if operation not in {"SUSPEND", "RESUME"}:
-            raise ValueError(f"Operation {operation} not supported for altering Dynamic Tables")
+        supported_target_properties = operation.supported_target_properties
+
+        if supported_target_properties:
+            if target_property is None or target_value is None:
+                raise ValueError(f"Target property and value must be provided for {operation.name} operation")
+
+            if target_property not in supported_target_properties:
+                raise ValueError(
+                    f"Only {', '.join(supported_target_properties)} supported as target property "
+                    f"for {operation.name} operation"
+                )
+
+        property_clause = f"{target_property}={target_value}" if target_property and target_value else ""
+        alter_momo_sql = (
+            f"""ALTER MODEL MONITOR {self._database_name}.{self._schema_name}.{monitor_name} """
+            f"""{operation.name} {property_clause}"""
+        )
+
         query_result_checker.SqlResultValidator(
             self._sql_client._session,
-            f"""ALTER MODEL MONITOR {self._database_name}.{self._schema_name}.{monitor_name} {operation}""",
+            alter_momo_sql,
             statement_params=statement_params,
         ).has_column("status").has_dimensions(1, 1).validate()
 
@@ -266,7 +321,7 @@ class ModelMonitorSQLClient:
         statement_params: Optional[dict[str, Any]] = None,
     ) -> None:
         self._alter_monitor(
-            operation="SUSPEND",
+            operation=MonitorOperation.SUSPEND,
             monitor_name=monitor_name,
             statement_params=statement_params,
         )
@@ -277,7 +332,37 @@ class ModelMonitorSQLClient:
         statement_params: Optional[dict[str, Any]] = None,
     ) -> None:
         self._alter_monitor(
-            operation="RESUME",
+            operation=MonitorOperation.RESUME,
+            monitor_name=monitor_name,
+            statement_params=statement_params,
+        )
+
+    def add_segment_column(
+        self,
+        monitor_name: sql_identifier.SqlIdentifier,
+        segment_column: sql_identifier.SqlIdentifier,
+        statement_params: Optional[dict[str, Any]] = None,
+    ) -> None:
+        """Add a segment column to the Model Monitor"""
+        self._alter_monitor(
+            operation=MonitorOperation.ADD,
+            monitor_name=monitor_name,
+            target_property="SEGMENT_COLUMN",
+            target_value=segment_column,
+            statement_params=statement_params,
+        )
+
+    def drop_segment_column(
+        self,
+        monitor_name: sql_identifier.SqlIdentifier,
+        segment_column: sql_identifier.SqlIdentifier,
+        statement_params: Optional[dict[str, Any]] = None,
+    ) -> None:
+        """Drop a segment column from the Model Monitor"""
+        self._alter_monitor(
+            operation=MonitorOperation.DROP,
             monitor_name=monitor_name,
+            target_property="SEGMENT_COLUMN",
+            target_value=segment_column,
             statement_params=statement_params,
         )

snowflake/ml/monitoring/_manager/model_monitor_manager.py

@@ -108,6 +108,7 @@ class ModelMonitorManager:
         prediction_class_columns = self._build_column_list_from_input(source_config.prediction_class_columns)
         actual_score_columns = self._build_column_list_from_input(source_config.actual_score_columns)
         actual_class_columns = self._build_column_list_from_input(source_config.actual_class_columns)
+        segment_columns = self._build_column_list_from_input(source_config.segment_columns)
 
         id_columns = [sql_identifier.SqlIdentifier(column_name) for column_name in source_config.id_columns]
         ts_column = sql_identifier.SqlIdentifier(source_config.timestamp_column)
@@ -123,6 +124,7 @@ class ModelMonitorManager:
             actual_score_columns=actual_score_columns,
             actual_class_columns=actual_class_columns,
             id_columns=id_columns,
+            segment_columns=segment_columns,
         )
 
         self._model_monitor_client.create_model_monitor(
@@ -144,6 +146,7 @@ class ModelMonitorManager:
             prediction_class_columns=prediction_class_columns,
             actual_score_columns=actual_score_columns,
             actual_class_columns=actual_class_columns,
+            segment_columns=segment_columns,
             refresh_interval=model_monitor_config.refresh_interval,
             aggregation_window=model_monitor_config.aggregation_window,
             baseline_database=baseline_database_name_id,

snowflake/ml/monitoring/entities/model_monitor_config.py

@@ -33,6 +33,9 @@ class ModelMonitorSourceConfig:
     baseline: Optional[str] = None
     """Name of table containing the baseline data."""
 
+    segment_columns: Optional[list[str]] = None
+    """List of columns in the source containing segment information for grouped monitoring."""
+
 
 @dataclass
 class ModelMonitorConfig:

snowflake/ml/monitoring/model_monitor.py

@@ -46,3 +46,29 @@ class ModelMonitor:
             telemetry.TelemetrySubProject.MONITORING.value,
         )
         self._model_monitor_client.resume_monitor(self.name, statement_params=statement_params)
+
+    @telemetry.send_api_usage_telemetry(
+        project=telemetry.TelemetryProject.MLOPS.value,
+        subproject=telemetry.TelemetrySubProject.MONITORING.value,
+    )
+    def add_segment_column(self, segment_column: str) -> None:
+        """Add a segment column to the Model Monitor"""
+        statement_params = telemetry.get_statement_params(
+            telemetry.TelemetryProject.MLOPS.value,
+            telemetry.TelemetrySubProject.MONITORING.value,
+        )
+        segment_column_id = sql_identifier.SqlIdentifier(segment_column)
+        self._model_monitor_client.add_segment_column(self.name, segment_column_id, statement_params=statement_params)
+
+    @telemetry.send_api_usage_telemetry(
+        project=telemetry.TelemetryProject.MLOPS.value,
+        subproject=telemetry.TelemetrySubProject.MONITORING.value,
+    )
+    def drop_segment_column(self, segment_column: str) -> None:
+        """Drop a segment column from the Model Monitor"""
+        statement_params = telemetry.get_statement_params(
+            telemetry.TelemetryProject.MLOPS.value,
+            telemetry.TelemetrySubProject.MONITORING.value,
+        )
+        segment_column_id = sql_identifier.SqlIdentifier(segment_column)
+        self._model_monitor_client.drop_segment_column(self.name, segment_column_id, statement_params=statement_params)

snowflake/ml/version.py

@@ -1,2 +1,2 @@
 # This is parsed by regex in conda recipe meta file. Make sure not to break it.
-VERSION = "1.11.0"
+VERSION = "1.12.0"