Skip to content

Commit db7d7d3

Browse files
authored
Disable Gremlin seed datasets for NA (#711)
1 parent 24853ea commit db7d7d3

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/graph_notebook/magics/graph_magic.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2881,8 +2881,8 @@ def seed(self, line, local_ns: dict = None):
28812881

28822882
if self.client.is_analytics_domain():
28832883
model_options = SEED_MODEL_OPTIONS_PG
2884-
custom_language_options = SEED_LANGUAGE_OPTIONS_PG
2885-
samples_pg_language_options = SEED_LANGUAGE_OPTIONS_PG
2884+
custom_language_options = SEED_LANGUAGE_OPTIONS_OC
2885+
samples_pg_language_options = SEED_LANGUAGE_OPTIONS_OC
28862886
else:
28872887
model_options = SEED_MODEL_OPTIONS
28882888
custom_language_options = SEED_LANGUAGE_OPTIONS
@@ -3158,11 +3158,7 @@ def process_gremlin_query_line(query_line, line_index, q):
31583158
logger.debug(f"Skipped blank query at line {line_index + 1} in seed file {q['name']}")
31593159
return 0
31603160
try:
3161-
if self.client.is_neptune_domain() and self.client.is_analytics_domain() and \
3162-
self.graph_notebook_config.gremlin.connection_protocol == DEFAULT_HTTP_PROTOCOL:
3163-
self.client.gremlin_http_query(query_line)
3164-
else:
3165-
self.client.gremlin_query(query_line)
3161+
self.client.gremlin_query(query_line)
31663162
return 0
31673163
except GremlinServerError as gremlinEx:
31683164
try:

0 commit comments

Comments
 (0)