diff --git a/pennylane/data/data_manager/__init__.py b/pennylane/data/data_manager/__init__.py index 198da52c1fb..c8c0dec51bc 100644 --- a/pennylane/data/data_manager/__init__.py +++ b/pennylane/data/data_manager/__init__.py @@ -309,7 +309,7 @@ def load( # pylint: disable=too-many-arguments Returns: list[:class:`~pennylane.data.Dataset`] - .. seealso:: :func:`~.load_interactive`, :func:`~.list_attributes`, :func:`~.list_datasets`. + .. seealso:: :func:`~.load_interactive`, :func:`~.list_attributes`, :func:`~.list_data_names`. **Example** @@ -439,7 +439,7 @@ def list_attributes(data_name) -> list[str]: Returns: list (str): A list of accepted attributes for a given data name - .. seealso:: :func:`~.load_interactive`, :func:`~.list_datasets`, :func:`~.load`. + .. seealso:: :func:`~.load_interactive`, :func:`~.list_data_names`, :func:`~.load`. **Example** @@ -553,7 +553,7 @@ def load_interactive(): **Example** - .. seealso:: :func:`~.load`, :func:`~.list_attributes`, :func:`~.list_datasets`. + .. seealso:: :func:`~.load`, :func:`~.list_attributes`, :func:`~.list_data_names`. .. code-block :: pycon @@ -617,7 +617,7 @@ def load_interactive(): __all__ = ( "load", "load_interactive", - "list_datasets", + "list_data_names", "list_attributes", "FULL", "DEFAULT", diff --git a/tests/data/data_manager/support.py b/tests/data/data_manager/support.py index 16261858395..fa4ba238fa3 100644 --- a/tests/data/data_manager/support.py +++ b/tests/data/data_manager/support.py @@ -1,90 +1,38 @@ """Test support for mocking GraphQL queries""" _list_attrs_resp = { - "data": { - "datasetClass": { - "attributes": [ - { - "name": "basis_rot_groupings" - }, - { - "name": "basis_rot_samples" - }, - { - "name": "dipole_op" - }, - { - "name": "fci_energy" - }, - { - "name": "fci_spectrum" - }, - { - "name": "hamiltonian" - }, - { - "name": "hf_state" - }, - { - "name": "molecule" - }, - { - "name": "number_op" - }, - { - "name": "optimal_sector" - }, - { - "name": "paulix_ops" - }, - { - "name": "qwc_groupings" - }, - { - "name": "qwc_samples" - }, - { - "name": "sparse_hamiltonian" - }, - { - "name": "spin2_op" - }, - { - "name": "spinz_op" - }, - { - "name": "symmetries" - }, - { - "name": "tapered_dipole_op" - }, - { - "name": "tapered_hamiltonian" - }, - { - "name": "tapered_hf_state" - }, - { - "name": "tapered_num_op" - }, - { - "name": "tapered_spin2_op" - }, - { - "name": "tapered_spinz_op" - }, - { - "name": "vqe_energy" - }, - { - "name": "vqe_gates" - }, - { - "name": "vqe_params" + "data": { + "datasetClass": { + "attributes": [ + {"name": "basis_rot_groupings"}, + {"name": "basis_rot_samples"}, + {"name": "dipole_op"}, + {"name": "fci_energy"}, + {"name": "fci_spectrum"}, + {"name": "hamiltonian"}, + {"name": "hf_state"}, + {"name": "molecule"}, + {"name": "number_op"}, + {"name": "optimal_sector"}, + {"name": "paulix_ops"}, + {"name": "qwc_groupings"}, + {"name": "qwc_samples"}, + {"name": "sparse_hamiltonian"}, + {"name": "spin2_op"}, + {"name": "spinz_op"}, + {"name": "symmetries"}, + {"name": "tapered_dipole_op"}, + {"name": "tapered_hamiltonian"}, + {"name": "tapered_hf_state"}, + {"name": "tapered_num_op"}, + {"name": "tapered_spin2_op"}, + {"name": "tapered_spinz_op"}, + {"name": "vqe_energy"}, + {"name": "vqe_gates"}, + {"name": "vqe_params"}, + ] } - ] } - } } _get_urls_resp = { diff --git a/tests/data/data_manager/test_dataset_access.py b/tests/data/data_manager/test_dataset_access.py index 512e0f479c5..2ffe46a6157 100644 --- a/tests/data/data_manager/test_dataset_access.py +++ b/tests/data/data_manager/test_dataset_access.py @@ -70,8 +70,6 @@ def graphql_mock(url, query, variables=None): """Return the JSON according to the query.""" if "ListAttributes" in query: json_data = _list_attrs_resp - elif "ListDatasets" in query: - json_data = _list_datasets_resp elif "GetDatasetsForDownload" in query: json_data = _get_urls_resp elif "GetParameterTree" in query: