Skip to content

Commit de0be55

Browse files
committed
CU-8699049kf: Update for use of convenience method for CDB/Vocab loading
1 parent c88dbe5 commit de0be55

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

medcat/1_create_model/create_modelpack/create_modelpack.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
from medcat.vocab import Vocab
44
from medcat.cdb import CDB
55
from medcat.cat import CAT
6-
from medcat.storage.serialisers import deserialise
76

87
# relative to file path
98
_FILE_DIR = os.path.dirname(__file__)
@@ -42,7 +41,7 @@ def load_cdb_and_save_modelpack(cdb_path: str,
4241
# Load cdb
4342
cdb: CDB
4443
try:
45-
cdb = deserialise(cdb_path)
44+
cdb = CDB.load(cdb_path)
4645
except NotADirectoryError:
4746
from medcat.utils.legacy.convert_cdb import get_cdb_from_old
4847
cdb = get_cdb_from_old(cdb_path)
@@ -61,7 +60,7 @@ def load_cdb_and_save_modelpack(cdb_path: str,
6160
# Load vocab
6261
vocab: Vocab
6362
try:
64-
vocab = deserialise(vocab_path)
63+
vocab = Vocab.load(vocab_path)
6564
except NotADirectoryError:
6665
from medcat.utils.legacy.convert_vocab import get_vocab_from_old
6766
vocab = get_vocab_from_old(vocab_path)

0 commit comments

Comments
 (0)