Skip to content

Commit 3e5a4b7

Browse files
committed
setting the service endpoint via env variable
1 parent 2c6d1f2 commit 3e5a4b7

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

tests/integration/feature_store/test_base.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,8 @@
2020
from ads.feature_store.statistics_config import StatisticsConfig
2121

2222

23-
client_kwargs = dict(
24-
retry_strategy=oci.retry.NoneRetryStrategy,
25-
service_endpoint=os.getenv("service_endpoint"),
26-
)
27-
ads.set_auth(client_kwargs=client_kwargs)
23+
ads.set_auth()
24+
os.environ["OCI_FS_SERVICE_ENDPOINT"] = os.getenv("service_endpoint")
2825

2926
try:
3027
from ads.feature_store.feature_store import FeatureStore
@@ -37,7 +34,9 @@
3734

3835
class FeatureStoreTestCase:
3936
# networks compartment in feature store
40-
TIME_NOW = str.format("{}_{}",datetime.utcnow().strftime("%Y_%m_%d_%H_%M_%S"),int(random()*1000))
37+
TIME_NOW = str.format(
38+
"{}_{}", datetime.utcnow().strftime("%Y_%m_%d_%H_%M_%S"), int(random() * 1000)
39+
)
4140
TENANCY_ID = "ocid1.tenancy.oc1..aaaaaaaa462hfhplpx652b32ix62xrdijppq2c7okwcqjlgrbknhgtj2kofa"
4241
COMPARTMENT_ID = "ocid1.tenancy.oc1..aaaaaaaa462hfhplpx652b32ix62xrdijppq2c7okwcqjlgrbknhgtj2kofa"
4342
METASTORE_ID = "ocid1.datacatalogmetastore.oc1.iad.amaaaaaabiudgxyap7tizm4gscwz7amu7dixz7ml3mtesqzzwwg3urvvdgua"

0 commit comments

Comments
 (0)