update tests

rstudio · Jan 3, 2024 · cc57572 · cc57572
1 parent 859748a
commit cc57572
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vetiver/tests/test_write_app.py b/vetiver/tests/test_write_app.py
@@ -26,9 +26,11 @@ def test_write_app(vetiver_model_creation):
         assert (
             contents
             == f"""from vetiver import VetiverModel
+from dotenv import load_dotenv, find_dotenv
 import vetiver
 import pins
 
+load_dotenv(find_dotenv())
 
 b = pins.board_folder({repr(tempdir)}, allow_pickle_read=True)
 v = VetiverModel.from_pin(b, 'model', version = {repr(version)})