Skip to content

Commit f20767f

Browse files
committed
fix tests
1 parent ba2429f commit f20767f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_dataset.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -332,7 +332,7 @@ def test_raises_error_for_duplicate():
332332
def test_dataset_export_autotag_tagged_items(CLIENT):
333333
# This test can only run for the test user who has an indexed dataset.
334334
# TODO: if/when we can create autotags via api, create one instead.
335-
if NUCLEUS_PYTEST_USER_ID in CLIENT.api_key:
335+
if NUCLEUS_PYTEST_USER_ID in os.environ["NUCLEUS_PYTEST_USER_ID"]:
336336
dataset = CLIENT.get_dataset(DATASET_WITH_AUTOTAG)
337337

338338
with pytest.raises(NucleusAPIError) as api_error:
@@ -362,7 +362,7 @@ def test_dataset_export_autotag_tagged_items(CLIENT):
362362
def test_dataset_export_autotag_training_items(CLIENT):
363363
# This test can only run for the test user who has an indexed dataset.
364364
# TODO: if/when we can create autotags via api, create one instead.
365-
if NUCLEUS_PYTEST_USER_ID in CLIENT.api_key:
365+
if NUCLEUS_PYTEST_USER_ID in os.environ["NUCLEUS_PYTEST_USER_ID"]:
366366
dataset = CLIENT.get_dataset(DATASET_WITH_AUTOTAG)
367367

368368
with pytest.raises(NucleusAPIError) as api_error:

0 commit comments

Comments
 (0)