Skip to content

Commit 0754ad8

Browse files
committed
better place for users to silence model card
1 parent bd30cc6 commit 0754ad8

File tree

2 files changed

+14
-12
lines changed

2 files changed

+14
-12
lines changed

vetiver/pin_read_write.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,10 @@
11
from .vetiver_model import VetiverModel
2+
from .utils import inform
23
import warnings
34
import logging
4-
import sys
5-
from types import SimpleNamespace
65

76
_log = logging.getLogger(__name__)
87

9-
modelcard_options = SimpleNamespace(quiet=False)
10-
11-
12-
def inform(log, msg):
13-
if log is not None:
14-
log.info(msg)
15-
16-
if not modelcard_options.quiet:
17-
print(msg, file=sys.stderr)
18-
198

209
class ModelCard(UserWarning):
2110
def __init__(

vetiver/utils.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
import nest_asyncio
22
import warnings
3+
import sys
4+
from types import SimpleNamespace
35

46
no_notebook = False
57
try:
@@ -18,3 +20,14 @@ def _jupyter_nb():
1820
nest_asyncio.apply()
1921
else:
2022
return False
23+
24+
25+
modelcard_options = SimpleNamespace(quiet=False)
26+
27+
28+
def inform(log, msg):
29+
if log is not None:
30+
log.info(msg)
31+
32+
if not modelcard_options.quiet:
33+
print(msg, file=sys.stderr)

0 commit comments

Comments
 (0)