Merge pull request #92 from rstudio/cleaning

Cleaning

Author: isabelizimm

.github/workflows/precommit.yml

@@ -0,0 +1,18 @@
+name: Code Checks
+
+on:
+  workflow_dispatch:
+  push:
+    branches: ['main']
+  pull_request:
+  release:
+    types: [published]
+
+jobs:
+  pre-commit:
+    name: "Run pre-commit"
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+      - uses: actions/setup-python@v2
+      - uses: pre-commit/action@v2.0.3

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 
 
-exclude: "(.*\\.csv)|(^pins/tests/_snapshots)"
+exclude: "(.*\\.csv)|(^examples/)|(^vetiver/tests/snapshots/)"
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v2.4.0

Makefile

@@ -20,6 +20,7 @@ help:
 	@echo "clean-test - remove test and coverage artifacts"
 	@echo "lint - check style with flake8"
 	@echo "test - run tests quickly with the default Python"
+	@echo "test-rsc - run tests for rsconnect"
 	@echo "coverage - check code coverage quickly with the default Python"
 	@echo "docs - generate Sphinx HTML documentation, including API docs"
 	@echo "cdocs - cleanout previous build & generate Sphinx HTML documentation, including API docs"
@@ -54,6 +55,9 @@ lint:
 	flake8 vetiver
 
 test: clean-test
+	pytest -m 'not rsc_test'
+
+test-rsc: clean-test
 	pytest
 
 coverage:

examples/coffeeratings.py

@@ -6,18 +6,32 @@
 from pathlib import Path
 
 # Load training data
-raw = pd.read_csv('https://raw.githubusercontent.com/rfordatascience/tidytuesday/master/data/2020/2020-07-07/coffee_ratings.csv')
+raw = pd.read_csv(
+    "https://raw.githubusercontent.com/rfordatascience/tidytuesday/master/data/2020/2020-07-07/coffee_ratings.csv"
+)
 df = pd.DataFrame(raw)
-coffee = df[["total_cup_points", "aroma", "flavor", "sweetness", "acidity", \
-    "body", "uniformity", "balance"]].dropna()
-
-X_train, X_test, y_train, y_test = model_selection.train_test_split(coffee.iloc[:,1:],coffee['total_cup_points'],test_size=0.2)
+coffee = df[
+    [
+        "total_cup_points",
+        "aroma",
+        "flavor",
+        "sweetness",
+        "acidity",
+        "body",
+        "uniformity",
+        "balance",
+    ]
+].dropna()
+
+X_train, X_test, y_train, y_test = model_selection.train_test_split(
+    coffee.iloc[:, 1:], coffee["total_cup_points"], test_size=0.2
+)
 
 # fit model
 lr_fit = LinearRegression().fit(X_train, y_train)
 
 # create vetiver model
-v = vetiver.VetiverModel(lr_fit, ptype_data=X_train, model_name = "v")
+v = vetiver.VetiverModel(lr_fit, ptype_data=X_train, model_name="v")
 
 # version model via pin
 from pins import board_folder
@@ -27,12 +41,12 @@
 model_board = board_folder(path=path, versioned=True, allow_pickle_read=True)
 vetiver_pin_write(board=model_board, model=v)
 
-myapp = vetiver.VetiverAPI(v, check_ptype = True)
+myapp = vetiver.VetiverAPI(v, check_ptype=True)
 api = myapp.app
 
 # next, run myapp.run() to start API and see visual documentation
 # create app.py file that includes pinned VetiverAPI to be deployed
-vetiver.vetiver_write_app(model_board, "v", file = path+"app.py")
+vetiver.vetiver_write_app(model_board, "v", file=path + "app.py")
 
 # automatically create requirements.txt
 vetiver.load_pkgs(model=v, path=path)

examples/coffeeratings/app.py

@@ -2,8 +2,8 @@
 import pins
 
 # edited to reflect Docker container path, and allow_pickle_read=True
-b = pins.board_folder('/code/app', allow_pickle_read=True)
-v = vetiver.vetiver_pin_read(b, 'v', version = '20220415T174503Z-06d9b')
+b = pins.board_folder("/code/app", allow_pickle_read=True)
+v = vetiver.vetiver_pin_read(b, "v", version="20220415T174503Z-06d9b")
 
 vetiver_api = vetiver.VetiverAPI(v)
 api = vetiver_api.app

examples/coffeeratings/vetiver_requirements.txt

@@ -72,4 +72,4 @@ uvicorn==0.17.6
     # via vetiver
 
 # manually edited to reflect appropriate versions of packages currently being developed
-git+https://github.com/isabelizimm/vetiver-python.git@37dff15a63238b1e6c44788af70473f19c29d83e
+git+https://github.com/isabelizimm/vetiver-python.git@37dff15a63238b1e6c44788af70473f19c29d83e

vetiver/__init__.py

@@ -14,8 +14,8 @@
 from .handlers.base import BaseHandler, create_handler, InvalidModelError  # noqa
 from .handlers.sklearn import SKLearnHandler  # noqa
 from .handlers.torch import TorchHandler  # noqa
-from .rsconnect import deploy_rsconnect # noqa
-from .monitor import compute_metrics, pin_metrics, plot_metrics, _rolling_df # noqa
+from .rsconnect import deploy_rsconnect  # noqa
+from .monitor import compute_metrics, pin_metrics, plot_metrics, _rolling_df  # noqa
 
 __author__ = "Isabel Zimmerman <isabel.zimmerman@rstudio.com>"
 __all__ = []

vetiver/data/__init__.py

@@ -12,6 +12,7 @@ def __dir__():
 
 def __getattr__(k):
     import pandas as pd
+
     f_path = sources.get("mtcars")
     if k == "chicago":
         f_path = sources.get("chicago")

vetiver/pin_read_write.py

@@ -61,12 +61,14 @@ def vetiver_pin_read(board, name: str, version: str = None) -> VetiverModel:
 
     Notes
     -----
-    If reading a board from RSConnect, the `board` argument must be in "username/modelname" format.
+    If reading a board from RSConnect, the `board` argument must be in
+    "username/modelname" format.
 
     """
 
     warnings.warn(
-        "vetiver_pin_read will be removed in v1.0.0. Use classmethod VetiverModel.from_pin() instead",
+        "vetiver_pin_read will be removed in v1.0.0. Use classmethod "
+        "VetiverModel.from_pin() instead",
         DeprecationWarning,
     )
 

vetiver/ptype.py

@@ -19,7 +19,7 @@ class NoAvailablePTypeError(Exception):
 
     def __init__(
         self,
-        message="There is no method available to create a 0-row input data prototype for `model`",
+        message="There is no method to create a 0-row input data prototype for `model`",
     ):
         self.message = message
         super().__init__(self.message)