diff --git a/setup.cfg b/setup.cfg index ef3a541..04c9f2e 100644 --- a/setup.cfg +++ b/setup.cfg @@ -32,10 +32,12 @@ install_requires = requests pins>=0.7.1 rsconnect-python>=1.11.0 + python-dotenv plotly pip-tools httpx + [options.extras_require] all = vetiver[dev] diff --git a/vetiver/tests/test_write_app.py b/vetiver/tests/test_write_app.py index 2c18727..cb07a17 100644 --- 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)}) diff --git a/vetiver/write_fastapi.py b/vetiver/write_fastapi.py index 116f14c..f378b23 100644 --- a/vetiver/write_fastapi.py +++ b/vetiver/write_fastapi.py @@ -104,7 +104,9 @@ def write_app( f = open(file, "x") app = f"""from vetiver import VetiverModel +from dotenv import load_dotenv, find_dotenv {_glue_required_pkgs(infra_pkgs)} +load_dotenv(find_dotenv()) b = pins.{load_board} {pin_read}