remove print statement

Author: sasha-scale

tests/test_dataset.py

@@ -30,7 +30,6 @@ def dataset(CLIENT):
 def test_dataset_create_and_delete(CLIENT):
     # Creation
     ds = CLIENT.create_dataset(TEST_DATASET_NAME)
-    print(ds)
     assert isinstance(ds, Dataset)
     assert ds.name == TEST_DATASET_NAME
     assert ds.model_runs == []
@@ -120,7 +119,10 @@ def test_slice_create_and_delete_and_list(dataset):
     assert response["dataset_id"] == dataset.id
     assert len(response["dataset_items"]) == 2
     for item in ds_items[:2]:
-        assert item.reference_id == response["dataset_items"][0]["ref_id"] or item.reference_id == response["dataset_items"][1]["ref_id"]
+        assert (
+            item.reference_id == response["dataset_items"][0]["ref_id"]
+            or item.reference_id == response["dataset_items"][1]["ref_id"]
+        )
 
 
 def test_slice_append(dataset):
@@ -149,4 +151,8 @@ def test_slice_append(dataset):
     print(response)
     assert len(response["dataset_items"]) == 3
     for item in ds_items[:3]:
-        assert item.reference_id == response["dataset_items"][0]["ref_id"] or item.reference_id == response["dataset_items"][1]["ref_id"] or item.reference_id == response["dataset_items"][2]["ref_id"]
+        assert (
+            item.reference_id == response["dataset_items"][0]["ref_id"]
+            or item.reference_id == response["dataset_items"][1]["ref_id"]
+            or item.reference_id == response["dataset_items"][2]["ref_id"]
+        )