Update differential_privacy/utils/budget.py

differential_privacy/utils/budget.py

@@ -10,6 +10,15 @@ def gen_rho(epsilon, delta):
     return (sqrt(epsilon+logterm) - sqrt(logterm))**2
 
 def date_to_str(year, month, day):
+    if isinstance(year, str):
+        year = int(year)
+    
+    if isinstance(month, str):
+        month = int(month)
+
+    if isinstance(day, str):
+        day = int(day)
+    
     return date(year=year, month=month, day=day).strftime("%Y-%m-%d")
 
 def write_to_global_privacy_budget(spark, table, epsilon, year, month, day, delta=0, rho=None):