Fixed bugs and integrated into new trapi_model version.

di2ag · May 10, 2021 · 067cf38 · 067cf38
1 parent fe4a6b2
commit 067cf38
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/chp_client/query.py b/chp_client/query.py
@@ -5,8 +5,9 @@
 import json
 from jsonschema import ValidationError
 
-from trapi_model import Message, Query
-from trapi_model.constants import *
+from trapi_model.query import Query
+from trapi_model.message import Message
+from trapi_model.biolink.constants import *
 
 from chp_client.exceptions import *
 

diff --git a/tests/generate_regression_tests.py b/tests/generate_regression_tests.py
@@ -3,7 +3,7 @@
 import random
 import sys
 
-from trapi_model.constants import *
+from trapi_model.biolink.constants import *
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)

diff --git a/tests/test_client.py b/tests/test_client.py
@@ -15,7 +15,7 @@
 import os
 from collections import defaultdict
 
-from trapi_model import Query
+from trapi_model.query import Query
 
 from generate_regression_tests import QueryBuildingRegressionSuite
 from chp_client.exceptions import QueryBuildError