diff --git a/docarray/store/jac.py b/docarray/store/jac.py index 88caaf84dd2..2ca4920194f 100644 --- a/docarray/store/jac.py +++ b/docarray/store/jac.py @@ -102,7 +102,9 @@ def list(namespace: str = '', show_table: bool = False) -> List[str]: from rich.table import Table resp = HubbleClient(jsonify=True).list_artifacts( - filter={'type': 'documentArray'}, sort={'createdAt': 1} + filter={'type': 'documentArray'}, + sort={'createdAt': 1}, + pageSize=10000, ) table = Table( @@ -267,10 +269,10 @@ def push_stream( # But it must be done this way for now because Hubble expects to know the length of the DocList # before it starts receiving the documents first_doc = next(docs) - docs = DocList[first_doc.__class__]([first_doc]) # type: ignore + _docs = DocList[first_doc.__class__]([first_doc]) # type: ignore for doc in docs: - docs.append(doc) - return cls.push(docs, name, public, show_progress, branding) + _docs.append(doc) + return cls.push(_docs, name, public, show_progress, branding) @staticmethod @hubble.login_required diff --git a/tests/integrations/store/test_jac.py b/tests/integrations/store/test_jac.py index 86baef1f801..87fd96f267d 100644 --- a/tests/integrations/store/test_jac.py +++ b/tests/integrations/store/test_jac.py @@ -13,8 +13,6 @@ TOLERANCE_RATIO = 0.5 # Percentage of difference allowed in stream vs non-stream test RANDOM: str = uuid.uuid4().hex[:8] -pytestmark = [pytest.mark.skip] - @pytest.fixture(scope='session', autouse=True) def testing_namespace_cleanup(): @@ -34,7 +32,7 @@ def testing_namespace_cleanup(): ) ) for da_name in da_names: - JACDocStore.delete(f'jac://{da_name}') + JACDocStore.delete(f'{da_name}') @pytest.mark.slow @@ -68,7 +66,6 @@ def test_pushpull_correct(capsys): assert len(captured.err) == 0, 'No error should be printed when show_progress=False' -@pytest.mark.skip(reason='time out') @pytest.mark.slow @pytest.mark.internet def test_pushpull_stream_correct(capsys): @@ -100,7 +97,6 @@ def test_pushpull_stream_correct(capsys): assert len(captured.err) == 0, 'No error should be printed when show_progress=False' -@pytest.mark.skip(reason='time out') @pytest.mark.slow @pytest.mark.internet def test_pull_stream_vs_pull_full(): @@ -157,7 +153,6 @@ def get_total_full(url: str): ), 'Full pull memory usage should be dependent on the size of the data' -@pytest.mark.skip(reason='The CI account might be broken') @pytest.mark.slow @pytest.mark.internet def test_list_and_delete(): @@ -214,7 +209,6 @@ def test_list_and_delete(): ), 'Deleting a non-existent DA should return False' -@pytest.mark.skip(reason='time out') @pytest.mark.slow @pytest.mark.internet def test_concurrent_push_pull():