Skip to content

Commit 7146807

Browse files
committed
logs, not warnings for model card
1 parent 133e91a commit 7146807

1 file changed

Lines changed: 25 additions & 2 deletions

File tree

vetiver/pin_read_write.py

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,22 @@
11
import warnings
22
from .vetiver_model import VetiverModel
33

4+
import logging
5+
import sys
6+
from types import SimpleNamespace
7+
8+
_log = logging.getLogger(__name__)
9+
10+
modelcard_options = SimpleNamespace(quiet=False)
11+
12+
13+
def inform(log, msg):
14+
if log is not None:
15+
log.info(msg)
16+
17+
if not modelcard_options.quiet:
18+
print(msg, file=sys.stderr)
19+
420

521
class ModelCard(UserWarning):
622
def __init__(
@@ -30,8 +46,15 @@ def vetiver_pin_write(board, model: VetiverModel, versioned: bool = True):
3046
if not board.allow_pickle_read:
3147
raise NotImplementedError # must be pickle-able
3248

33-
warnings.simplefilter("once", ModelCard)
34-
warnings.warn(ModelCard().message)
49+
# warnings.simplefilter("once", ModelCard)
50+
# warnings.warn(ModelCard().message)
51+
52+
inform(
53+
_log,
54+
"Model Cards provide a framework for transparent, responsible "
55+
"reporting. \n Use the vetiver `.qmd` Quarto template as a place to start, \n "
56+
"with vetiver.model_card()",
57+
)
3558

3659
board.pin_write(
3760
model.model,

0 commit comments

Comments
 (0)