Skip to content

Commit a87f77d

Browse files
committed
fix: use ServerZone class for remote evaluation server_zone config
1 parent f223bce commit a87f77d

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

src/amplitude_experiment/remote/config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
from src.amplitude_experiment import ServerZone
2+
13
DEFAULT_SERVER_URL = 'https://api.lab.amplitude.com'
24
EU_SERVER_URL = 'https://api.lab.eu.amplitude.com'
35

@@ -12,7 +14,7 @@ def __init__(self, debug=False,
1214
fetch_retry_backoff_max_millis=10000,
1315
fetch_retry_backoff_scalar=1.5,
1416
fetch_retry_timeout_millis=10000,
15-
server_zone='us'):
17+
server_zone: ServerZone = ServerZone.US):
1618
"""
1719
Initialize a config
1820
Parameters:
@@ -41,6 +43,6 @@ def __init__(self, debug=False,
4143
self.fetch_retry_backoff_scalar = fetch_retry_backoff_scalar
4244
self.fetch_retry_timeout_millis = fetch_retry_timeout_millis
4345
self.server_zone = server_zone
44-
if server_url == DEFAULT_SERVER_URL and server_zone == 'eu':
46+
if server_url == DEFAULT_SERVER_URL and server_zone == ServerZone.EU:
4547
self.server_url = EU_SERVER_URL
4648

0 commit comments

Comments
 (0)