fix tests

Author: isabelizimm

vetiver/__init__.py

@@ -19,4 +19,4 @@
 from .handlers.sklearn import SKLearnHandler  # noqa
 from .handlers.torch import TorchHandler  # noqa
 from .rsconnect import deploy_rsconnect # noqa
-from .monitor import compute_metrics, pin_metrics, plot_metrics # noqa
+from .monitor import compute_metrics, pin_metrics, plot_metrics, _rolling_df # noqa

vetiver/tests/snapshots/test_monitor.json

@@ -0,0 +1 @@
+{"index":{"0":1325376000000,"1":1325376000000,"2":1325376002000,"3":1325376002000,"4":1325376004000,"5":1325376004000,"6":1325376006000,"7":1325376006000,"8":1325376008000,"9":1325376008000},"n":{"0":2,"1":2,"2":2,"3":2,"4":2,"5":2,"6":2,"7":2,"8":2,"9":2},"metric":{"0":"mean_squared_error","1":"mean_absolute_error","2":"mean_squared_error","3":"mean_absolute_error","4":"mean_squared_error","5":"mean_absolute_error","6":"mean_squared_error","7":"mean_absolute_error","8":"mean_squared_error","9":"mean_absolute_error"},"estimate":{"0":0.0,"1":0.0,"2":0.0,"3":0.0,"4":0.0,"5":0.0,"6":0.0,"7":0.0,"8":0.0,"9":0.0}}

vetiver/tests/test_monitor.py

@@ -1,4 +1,3 @@
-from typing import List
 from sklearn import metrics
 from datetime import timedelta
 import pandas as pd
@@ -29,7 +28,7 @@ def test_compute():
     )
 
 def test_monitor(snapshot):
-    snapshot.snapshot_dir = './snapshots'
+    snapshot.snapshot_dir = './vetiver/tests/snapshots'
     m = vetiver.compute_metrics(
         df, "index", td, metric_set=metric_set, truth="x", estimate="y"
     )