format

PennyLaneAI · Sep 19, 2024 · b7e2544 · b7e2544
1 parent dfcceab
commit b7e2544
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 88 deletions.
diff --git 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
@@ -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
@@ -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: