fix: Fix source code to properly filter using dashboard variables and modify test_dashboard_methods to take into account the new filters in the dashboard variables

Author: Sushanth Sathish Kumar

src/sagemaker/dashboard/data_quality_dashboard.py

@@ -120,7 +120,7 @@ def _generate_variables(self):
                     variable_id="Feature",
                     label="Feature",
                     search=self.DATA_QUALITY_METRICS_BATCH_NAMESPACE
-                    + f'MonitoringSchedule="{self.monitoring_schedule}" ',
+                    + f' MonitoringSchedule="{self.monitoring_schedule}" ',
                     populateFrom="Feature",
                 )
             ]
@@ -133,8 +133,8 @@ def _generate_variables(self):
                 variable_id="Feature",
                 label="Feature",
                 search=self.DATA_QUALITY_METRICS_ENDPOINT_NAMESPACE
-                + f'Endpoint="{self.endpoint}" '
-                + f'MonitoringSchedule="{self.monitoring_schedule}" ',
+                + f' Endpoint="{self.endpoint}"'
+                + f' MonitoringSchedule="{self.monitoring_schedule}" ',
                 populateFrom="Feature",
             )
         ]

tests/unit/test_dashboard_methods.py

@@ -118,7 +118,9 @@ def test_automatic_data_quality_dashboard_endpoint():
             inputType="select",
             variable_id="Feature",
             label="Feature",
-            search="{aws/sagemaker/Endpoints/data-metrics,Endpoint,Feature,MonitoringSchedule}",
+            search="{aws/sagemaker/Endpoints/data-metrics,Endpoint,Feature,MonitoringSchedule}"
+            + ' Endpoint="endpoint"'
+            + ' MonitoringSchedule="monitoring_schedule" ',
             populateFrom="Feature",
         )
     ]
@@ -259,7 +261,7 @@ def test_automatic_data_quality_dashboard_endpoint():
             ]
         ],
     }
-
+    
     assert dashboard.to_dict() == expected_dashboard
 
 
@@ -271,7 +273,8 @@ def test_automatic_data_quality_dashboard_batch_transform():
             inputType="select",
             variable_id="Feature",
             label="Feature",
-            search="{aws/sagemaker/ModelMonitoring/data-metrics,Feature,MonitoringSchedule}",
+            search="{aws/sagemaker/ModelMonitoring/data-metrics,Feature,MonitoringSchedule}"
+            + ' MonitoringSchedule="monitoring_schedule" ',
             populateFrom="Feature",
         )
     ]