Skip to content

Commit 7dafba4

Browse files
committed
Set new env variable for starting session docker image
1 parent f30afb9 commit 7dafba4

File tree

1 file changed

+2
-1
lines changed
  • graphdatascience/tests/integrationV2/procedure_surface

1 file changed

+2
-1
lines changed

graphdatascience/tests/integrationV2/procedure_surface/conftest.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,13 +74,14 @@ def start_session(
7474
.with_env("ALLOW_LIST", "DEFAULT")
7575
.with_env("DNS_NAME", "gds-session")
7676
.with_env("PAGE_CACHE_SIZE", "100M")
77+
.with_env("MODEL_STORAGE_BASE_LOCATION", "/models")
7778
.with_exposed_ports(8491)
7879
.with_volume_mapping(password_dir, "/passwords")
7980
)
8081
if not inside_ci():
8182
session_container = session_container.with_network(network).with_network_aliases("gds-session")
8283
with session_container as session_container:
83-
wait_for_logs(session_container, "Running GDS tasks: 0")
84+
wait_for_logs(session_container, "Running GDS tasks: 0", timeout=20)
8485
yield GdsSessionConnectionInfo(
8586
host=session_container.get_container_host_ip(),
8687
arrow_port=session_container.get_exposed_port(8491),

0 commit comments

Comments
 (0)