Skip to content

Commit 7447ffd

Browse files
committed
tweak test for precommit
1 parent 3497525 commit 7447ffd

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

vetiver/tests/test_rsconnect.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,12 @@
33
import sklearn
44
from pins.boards import BoardRsConnect
55
from pins.rsconnect.api import RsConnectApi
6+
from pins.rsconnect.fs import RsConnectFs
67
from vetiver import VetiverModel, vetiver_pin_write, mock
78
import sklearn
9+
from rsconnect.api import RSConnectServer
810

911
import vetiver
10-
from vetiver.rsconnect import deploy_rsconnect
1112

1213
# Load data, model
1314
X_df, y = vetiver.mock.get_mock_data()
@@ -20,14 +21,11 @@
2021

2122

2223
def server_from_key(name):
23-
from rsconnect.api import RSConnectServer
24-
2524
with open(RSC_KEYS_FNAME) as f:
2625
api_key = json.load(f)[name]
2726
return RSConnectServer(RSC_SERVER_URL, api_key)
2827

2928
def rsc_from_key(name):
30-
3129
with open(RSC_KEYS_FNAME) as f:
3230
api_key = json.load(f)[name]
3331
return RsConnectApi(RSC_SERVER_URL, api_key)
@@ -79,7 +77,7 @@ def test_deploy(rsc_short):
7977

8078
vetiver.vetiver_pin_write(board=rsc_short, model=v)
8179

82-
deploy_rsconnect(
80+
vetiver.deploy_rsconnect(
8381
connect_server=server_from_key("susan"),
8482
board=rsc_short,
8583
pin_name="susan/model"

0 commit comments

Comments
 (0)