make enum key to be snake_case

whylabs · Sep 18, 2023 · c940581 · c940581
1 parent ffb915c
commit c940581
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/monitor/manager/test_monitor_setup.py b/tests/monitor/manager/test_monitor_setup.py
@@ -246,12 +246,12 @@ def test_dataset_metrics_are_warned_on_setup(caplog):
 def test_dataset_matrix_if_metric_is_missing_datapoint(monitor_setup) -> None:
     monitor_setup.config = FixedThresholdsConfig(
         upper=0,
-        metric=DatasetMetric.missingDataPoint
+        metric=DatasetMetric.missing_data_point
     )
     monitor_setup.data_readiness_duration = "P1DT18H"
     monitor_setup.apply()
 
-    assert monitor_setup.analyzer.config.metric == DatasetMetric.missingDataPoint
+    assert monitor_setup.analyzer.config.metric == DatasetMetric.missing_data_point
     assert monitor_setup.analyzer.dataReadinessDuration == "P1DT18H"
     assert isinstance(monitor_setup.analyzer.targetMatrix, DatasetMatrix)
 

diff --git a/whylabs_toolkit/monitor/models/analyzer/algorithms.py b/whylabs_toolkit/monitor/models/analyzer/algorithms.py
@@ -59,7 +59,7 @@ class DatasetMetric(str, Enum):
     regression_rmse = "regression.rmse"
 
     # other metrics
-    missingDataPoint = "missingDatapoint"
+    missing_data_point = "missingDatapoint"
 
 
 class SimpleColumnMetric(str, Enum):