11import pins
2+ import warnings
3+
24from .vetiver_model import VetiverModel
35from .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