From 067cf3841101b1c80bd246dfbe104f8b6f3274ea Mon Sep 17 00:00:00 2001
From: Chase Yakaboski <chase.a.yakaboski@dartmouth.edu>
Date: Sun, 9 May 2021 22:31:17 -0400
Subject: [PATCH] Fixed bugs and integrated into new trapi_model version.

---
 chp_client/query.py                | 5 +++--
 tests/generate_regression_tests.py | 2 +-
 tests/test_client.py               | 2 +-
 3 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/chp_client/query.py b/chp_client/query.py
index eb28d4c..5375be1 100644
--- 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
index d1b0cb5..b077356 100644
--- 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
index 342532a..d0b2266 100644
--- 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