Merge pull request #89 from isabelizimm/date-monitor

coerce date var to datetime

Author: isabelizimm

vetiver/monitor.py

@@ -1,5 +1,6 @@
 import plotly.express as px
 import pandas as pd
+import numpy as np
 from datetime import timedelta
 
 
@@ -41,7 +42,12 @@ def compute_metrics(
 
     """
 
-    df = data[[truth, estimate, date_var]].set_index(date_var).sort_index()
+    df = data[[truth, estimate, date_var]].copy()
+
+    if not np.issubdtype(df[date_var], np.datetime64):
+        df[date_var] = pd.to_datetime(df[date_var])
+
+    df = df.set_index(date_var).sort_index()
     lst = [_ for _ in _rolling_df(df=df, td=period)]
 
     rows = []
@@ -169,7 +175,7 @@ def plot_metrics(
         color=metric,
         facet_row=metric,
         markers=dict(size=n),
-        hover_data={"n": ':'},
+        hover_data={"n": ":"},
         **kw,
     )
 

vetiver/tests/test_monitor.py

@@ -35,6 +35,27 @@ def test_compute():
     )
 
 
+def test_compute_coerce_datetime():
+
+    df_metrics = pd.DataFrame(
+        {
+            "index": ["2021-01-01", "2021-01-02", "2021-01-03"],
+            "truth": [200, 201, 199],
+            "pred": [198, 200, 199],
+        }
+    )
+    td = timedelta(days=1)
+    m = vetiver.compute_metrics(
+        df_metrics, "index", td, metric_set=metric_set, truth="truth", estimate="pred"
+    )
+    assert isinstance(m, pd.DataFrame)
+    assert m.shape == (4, 4)
+    numpy.testing.assert_array_equal(
+        m.metric.unique(),
+        numpy.array(["mean_squared_error", "mean_absolute_error"], dtype=object),
+    )
+
+
 def test_monitor(snapshot):
     snapshot.snapshot_dir = "./vetiver/tests/snapshots"
     m = vetiver.compute_metrics(