Skip to content

Commit 4b3ff57

Browse files
Move bayesdb_metamodel_version to src/metamodel.py.
1 parent 5ea0b5b commit 4b3ff57

File tree

2 files changed

+9
-6
lines changed

2 files changed

+9
-6
lines changed

src/metamodel.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,8 @@
4040
print x
4141
"""
4242

43+
from bayeslite.util import cursor_value
44+
4345
builtin_metamodels = []
4446
builtin_metamodel_names = set()
4547

@@ -74,6 +76,12 @@ def bayesdb_deregister_metamodel(bdb, metamodel):
7476
assert bdb.metamodels[name] == metamodel
7577
del bdb.metamodels[name]
7678

79+
def bayesdb_metamodel_version(bdb, mm_name):
80+
cursor = bdb.sql_execute('''
81+
SELECT version FROM bayesdb_metamodel WHERE name = ?
82+
''', (mm_name,))
83+
return cursor_value(cursor, nullok=True)
84+
7785
class IBayesDBMetamodel(object):
7886
"""BayesDB metamodel interface.
7987

src/metamodels/cgpm_metamodel.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@
6161

6262
from bayeslite.exception import BQLError
6363
from bayeslite.metamodel import IBayesDBMetamodel
64+
from bayeslite.metamodel import bayesdb_metamodel_version
6465
from bayeslite.sqlite3_util import sqlite3_quote_name
6566
from bayeslite.stats import arithmetic_mean
6667
from bayeslite.util import casefold
@@ -785,11 +786,5 @@ def _default_numerical(bdb, generator_id, var):
785786

786787
# XXX Move these utilities elsewhere.
787788

788-
def bayesdb_metamodel_version(bdb, mm_name):
789-
cursor = bdb.sql_execute('''
790-
SELECT version FROM bayesdb_metamodel WHERE name = ?
791-
''', (mm_name,))
792-
return cursor_value(cursor, nullok=True)
793-
794789
def json_dumps(obj):
795790
return json.dumps(obj, sort_keys=True)

0 commit comments

Comments
 (0)