Skip to content

Commit

Permalink
Update test_client.py (#83)
Browse files Browse the repository at this point in the history
* Update test_client.py

* Update test_client.py

* Update test_client.py
  • Loading branch information
jonathanfeng-scale committed Jan 18, 2024
1 parent c9775e1 commit 39e4613
Showing 1 changed file with 0 additions and 22 deletions.
22 changes: 0 additions & 22 deletions tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -498,25 +498,3 @@ def test_invite_teammates():
assert len(new_teammates) >= len(
old_teammates
) # needs to sleep for teammates list to be updated


STUDIO_TEST_PROJECT = "python-sdk-studio-test"

try:
project = client.get_project(STUDIO_TEST_PROJECT)
except ScaleResourceNotFound:
client.create_project(
project_name=STUDIO_TEST_PROJECT, task_type=TaskType.ImageAnnotation
)
STUDIO_BATCH_TEST_NAME = f"studio-test-batch-{current_timestamp}"


def test_list_studio_batches():
# Create a test project if it does not already exist

# Create a test batch
client.create_batch(STUDIO_TEST_PROJECT, STUDIO_BATCH_TEST_NAME)

# Check that the batch is returned by the list_studio_batches method
batches = client.list_studio_batches()
assert len(batches) > 0

0 comments on commit 39e4613

Please sign in to comment.