make pytest happy

isabelizimm · isabelizimm · commit e325c3ef54f0 · 2022-07-05T17:53:41.000-04:00
diff --git a/vetiver/data/__init__.py b/vetiver/data/__init__.py
@@ -12,7 +12,10 @@ def __dir__():
 
 def __getattr__(k):
     import pandas as pd
-
-    f_path = sources.get(k)
+    f_path = None
+    if k == "mtcars":
+        f_path = sources.get("mtcars")
+    elif k == "chicago":
+        f_path = sources.get("chicago")
 
     return pd.read_csv(f_path)
