add seed to model script

Author: isabelizimm

.github/workflows/tests.yml

@@ -35,7 +35,7 @@ jobs:
           python -m pip install -e .[dev,torch,statsmodels,xgboost]
       - name: Run Tests
         run: |
-          pytest -m 'not rsc_test' --cov --cov-report xml
+          pytest -m 'not rsc_test and not docker' --cov --cov-report xml
       - name: Upload coverage
         uses: codecov/codecov-action@v2
 

script/setup-docker/docker.py

@@ -1,5 +1,8 @@
 import vetiver
 import pins
+import numpy as np
+
+np.random.seed(500)
 
 X, y = vetiver.get_mock_data()
 model = vetiver.get_mock_model().fit(X, y)

vetiver/tests/test_prepare_docker.py

@@ -6,14 +6,14 @@
 DOCKER_URL = "http://0.0.0.0:8080/predict"
 
 pytestmark = pytest.mark.docker  # noqa
-np.random.seed(500)
 
 
 def test_predict_sklearn_df_check_ptype():
+    np.random.seed(500)
 
     X, y = vetiver.mock.get_mock_data()
     response = vetiver.predict(endpoint=DOCKER_URL, data=X)
 
     assert isinstance(response, pd.DataFrame), response
-    assert response.iloc[0, 0] == 54.52
+    assert response.iloc[0, 0] == 44.47
     assert len(response) == 100

vetiver/write_docker.py

@@ -1,5 +1,6 @@
 import sys
 import warnings
+from pathlib import Path
 
 from .write_fastapi import write_app
 from .attach_pkgs import load_pkgs
@@ -92,14 +93,14 @@ def write_docker(
 CMD ["uvicorn", "app.app:api", "--host", "{host}", "--port", "{port}"]
 """
 
-    f = open(f"{path}Dockerfile", "x")
+    f = open(Path(path, "Dockerfile"), "x")
     f.write(docker_script)
 
 
 def prepare_docker(
     board,
     pin_name: str,
-    path: str = "",
+    path: str = "./",
     version=None,
     rspm_env: bool = False,
     host: str = "0.0.0.0",
@@ -125,8 +126,10 @@ def prepare_docker(
             Port to run VetiverAPI from Dockerfile
 
     """
-    # urljoin everything
+
     v = VetiverModel.from_pin(board=board, name=pin_name, version=version)
-    write_app(board=board, pin_name=pin_name, version=version, file=f"{path}app.py")
-    load_pkgs(v, path=f"{path}vetiver_")
+    write_app(
+        board=board, pin_name=pin_name, version=version, file=Path(path, "app.py")
+    )
+    load_pkgs(v, path=Path(path, "vetiver_"))
     write_docker(path=path, rspm_env=rspm_env, host=host, port=port)