Merge pull request #390 from DoubleML/sk-update-did-jitter-plot

fix: ensure jittered_x is correctly initialized as datetime

Author: SvenKlaassen

doubleml/did/utils/_plot.py

@@ -26,7 +26,7 @@ def add_jitter(data, x_col, is_datetime=None, jitter_value=None):
 
     # Initialize jittered_x with original values
     if is_datetime:
-        data["jittered_x"] = data[x_col]
+        data["jittered_x"] = pd.to_datetime(data[x_col]).astype("datetime64[ns]")
     else:
         data["jittered_x"] = data[x_col].astype(float)