Add alias iia for identity initialised ansatz

pasqal-io · Nov 27, 2024 · 508f7c2 · 508f7c2
1 parent 8e8ae65
commit 508f7c2
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git a/qadence/constructors/__init__.py b/qadence/constructors/__init__.py
@@ -7,8 +7,7 @@
 
 from .hea import hea
 from .alt import alt
-
-from .iia import identity_initialized_ansatz
+from .iia import identity_initialized_ansatz, iia
 
 from .daqc import daqc_transform
 
@@ -32,6 +31,7 @@
     "hea",
     "alt",
     "identity_initialized_ansatz",
+    "iia",
     "hamiltonian_factory",
     "ising_hamiltonian",
     "ObservableConfig",

diff --git a/qadence/constructors/iia.py b/qadence/constructors/iia.py
@@ -240,3 +240,6 @@ def identity_initialized_ansatz(
         initialized_layers.append(tag(chain(*krons), tag=f"BPMA-{layer}"))
 
     return chain(*initialized_layers)
+
+
+iia = identity_initialized_ansatz
diff --git a/qadence/ml_tools/constructors.py b/qadence/ml_tools/constructors.py
@@ -13,7 +13,7 @@
     analog_feature_map,
     feature_map,
     hamiltonian_factory,
-    identity_initialized_ansatz,
+    iia,
     rydberg_feature_map,
     rydberg_hea,
     rydberg_tower_feature_map,
@@ -414,7 +414,7 @@ def _create_iia_digital(
     entangler = config.strategy_args.get("entangler", CNOT)
     periodic = config.strategy_args.get("periodic", False)
 
-    return identity_initialized_ansatz(
+    return iia(
         n_qubits=num_qubits,
         depth=config.depth,
         param_prefix=config.param_prefix,
@@ -443,7 +443,7 @@ def _create_iia_sdaqc(
     entangler = config.strategy_args.get("entangler", CNOT)
     periodic = config.strategy_args.get("periodic", False)
 
-    return identity_initialized_ansatz(
+    return iia(
         n_qubits=num_qubits,
         depth=config.depth,
         param_prefix=config.param_prefix,