Skip to content

Commit 18a6325

Browse files
committed
allow user to set version
1 parent c1c1756 commit 18a6325

2 files changed

Lines changed: 15 additions & 7 deletions

File tree

setup.cfg

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ install_requires =
3333
nest-asyncio
3434
torch
3535
requests
36+
git+https://github.com/machow/pins-python
3637

3738
[options.extras_require]
3839
dev =

vetiver/pin_read_write.py

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
import pins
2+
import warnings
3+
24
from .vetiver_model import VetiverModel
35
from .meta import vetiver_meta
46

5-
def vetiver_pin_write(board, model: VetiverModel):
7+
def vetiver_pin_write(board, model: VetiverModel, versioned: bool=True):
68
"""
79
Write pin including VetiverModel
810
@@ -22,25 +24,30 @@ def vetiver_pin_write(board, model: VetiverModel):
2224
type = "joblib",
2325
description = model.description,
2426
metadata = {"required_pkgs": model.metadata.get("required_pkgs"),
25-
"ptype": None if model.ptype == None else model.ptype().json()}
27+
"ptype": None if model.ptype == None else model.ptype().json()},
28+
versioned=versioned
2629
)
27-
#return(print("create a model card for your published model"))
2830

29-
# """Create a Model Card for your published model.
31+
# to do: Model card
32+
33+
# message = """
34+
# Create a Model Card for your published model.
3035
# Model Cards provide a framework for transparent, responsible reporting.
3136
# Use the vetiver `.Rmd` template as a place to start."""
3237

38+
# warnings.warn(message=message)
39+
3340

34-
def vetiver_pin_read(board, name) -> VetiverModel:
41+
def vetiver_pin_read(board: pins.BaseBoard, name: str, version: str = None) -> VetiverModel:
3542

36-
model = board.pin_read(name)
43+
model = board.pin_read(name, version)
3744
meta = board.pin_meta(name)
3845

3946
v = VetiverModel(model = model,
4047
model_name = name,
4148
description = meta.description,
4249
metadata = vetiver_meta(user = meta.user,
43-
version = meta.pin_hash,
50+
version = version,
4451
url = meta.user.get("url"), # None all the time, besides Connect
4552
required_pkgs = meta.user.get("required_pkgs")
4653
),

0 commit comments

Comments
 (0)