Merge branch 'main' into fixing-local-detuning-braket-schema

QuEraComputing · Oct 23, 2023 · 00a3370 · 00a3370
2 parents 83b5f42 + 624daff
commit 00a3370
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,7 +4,7 @@ out.txt
 .DS_Store
 tests/data/jobs/
 main.html
-main*.py
+main
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

diff --git a/src/bloqade/__init__.py b/src/bloqade/__init__.py
@@ -11,6 +11,7 @@
 )
 import bloqade.ir as _ir
 from bloqade.constants import RB_C6
+from bloqade.submission.capabilities import get_capabilities
 
 import importlib.metadata
 
@@ -52,4 +53,5 @@ def tree_depth(depth: int = None):
     "dumps",
     "rydberg_h",
     "waveform",
+    "get_capabilities",
 ]
diff --git a/src/bloqade/submission/capabilities.py b/src/bloqade/submission/capabilities.py
@@ -4,7 +4,12 @@
 
 
 # TODO: Create unit converter for capabilities
-def get_capabilities():
+def get_capabilities() -> QuEraCapabilities:
+    """Get the device capabilities for Aquila
+
+    Returns:
+        QuEraCapabilities: capabilities object for Aquila
+    """
     base_path = os.path.dirname(__file__)
     full_path = os.path.join(
         base_path, "quera_api_client", "config", "capabilities.json"