Skip to content

Commit 4df3530

Browse files
committed
upate pin and model
1 parent cfc1992 commit 4df3530

2 files changed

Lines changed: 2 additions & 13 deletions

File tree

vetiver/pin_read_write.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,6 @@
11
import warnings
22

33
from .vetiver_model import VetiverModel
4-
from .write_fastapi import _choose_version
54

65

76
def vetiver_pin_write(board, model: VetiverModel, versioned: bool = True):
@@ -73,10 +72,6 @@ def vetiver_pin_read(board, name: str, version: str = None) -> VetiverModel:
7372
DeprecationWarning,
7473
)
7574

76-
version = (
77-
version if version is not None else _choose_version(board.pin_versions(name))
78-
)
79-
8075
v = VetiverModel.from_pin(board=board, name=name, version=version)
8176

8277
return v

vetiver/vetiver_model.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
from vetiver.handlers.base import create_handler
44
from .meta import _model_meta
5-
from .write_fastapi import _choose_version
65

76

87
class NoModelAvailableError(Exception):
@@ -78,22 +77,17 @@ def __init__(
7877

7978
@classmethod
8079
def from_pin(cls, board, name: str, version: str = None):
81-
version = (
82-
version
83-
if version is not None
84-
else _choose_version(board.pin_versions(name))
85-
)
8680

8781
model = board.pin_read(name, version)
88-
meta = board.pin_meta(name)
82+
meta = board.pin_meta(name, version)
8983

9084
return cls(
9185
model=model,
9286
model_name=name,
9387
description=meta.description,
9488
metadata=_model_meta(
9589
user=meta.user,
96-
version=version,
90+
version=meta.version.version.abc,
9791
url=meta.user.get("url"), # None all the time, besides Connect
9892
required_pkgs=meta.user.get("required_pkgs"),
9993
),

0 commit comments

Comments
 (0)