simplify cleaning drivers

dyvenia · Nov 9, 2023 · b46b3e9 · b46b3e9
1 parent c9c3c50
commit b46b3e9
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/viadot/tasks/customer_gauge.py b/viadot/tasks/customer_gauge.py
@@ -313,13 +313,9 @@ def _drivers_cleaner(
             str: A cleaned and formatted string of driver data.
         """
 
-        drivers = drivers.split("}, {")
-        cleaned_drivers = []
-        for driver in drivers:
-            driver = driver.replace("{", "").replace("}", "")
-            driver = driver.replace("'", "").replace("label: ", "")
-            cleaned_drivers.append(driver)
-        return ', '.join(cleaned_drivers)  
+        cleaned_drivers = drivers.replace("{", "").replace("}", "").replace("'", "").replace("label: ", "")
+
+        return cleaned_drivers
 
     def __call__(self):
         """Download Customer Gauge data to a DF"""