Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 20 additions & 0 deletions .github/workflows/tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,26 @@ jobs:
run: |
pytest vetiver -m 'rsc_test'

test-docker:
name: "Test Docker"
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- uses: actions/setup-python@v2
with:
python-version: 3.8
- name: Install dependencies
run: |
python -m pip install --upgrade pip
python -m pip install -e ".[dev]"
- name: run Docker
run: |
python script/setup-docker/docker.py
docker build -t mock .
docker run -d -v $PWD/pinsboard:/vetiver/pinsboard -p 8080:8080 mock
sleep 5
curl -s --retry 10 --retry-connrefused http://0.0.0.0:8080

test-no-extras:
name: "Test no exra ml frameworks"
runs-on: ubuntu-latest
Expand Down
4 changes: 4 additions & 0 deletions script/setup-docker/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
vetiver_requirements.txt
app.py
Dockerfile
pinsboard
14 changes: 14 additions & 0 deletions script/setup-docker/docker.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
import vetiver
import pins

X, y = vetiver.get_mock_data()
model = vetiver.get_mock_model().fit(X, y)

board = pins.board_folder("pinsboard", allow_pickle_read=True)

v = vetiver.VetiverModel(model, "mymodel", ptype_data=X)

vetiver.vetiver_pin_write(board, v)
vetiver.load_pkgs(v, path="vetiver_")
vetiver.write_app(board, "mymodel")
vetiver.write_docker()
4 changes: 2 additions & 2 deletions vetiver/tests/test_write_docker.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ def test_vetiver_write_app_no_rspm():
RUN pip install --no-cache-dir --upgrade -r /vetiver/requirements.txt

# copy app file
COPY app.py /vetiver/app
COPY app.py /vetiver/app/app.py

# expose port
EXPOSE 8080
Expand Down Expand Up @@ -64,7 +64,7 @@ def test_vetiver_write_app_rspm():
RUN pip install --no-cache-dir --upgrade -r /vetiver/requirements.txt

# copy app file
COPY app.py /vetiver/app
COPY app.py /vetiver/app/app.py

# expose port
EXPOSE 8080
Expand Down
2 changes: 1 addition & 1 deletion vetiver/write_docker.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,7 +79,7 @@ def write_docker(
RUN pip install --no-cache-dir --upgrade -r /vetiver/requirements.txt

# copy app file
COPY {app_file} /vetiver/app
COPY {app_file} /vetiver/app/{app_file}

# expose port
EXPOSE {port}
Expand Down