Bugfix for models from pspecs with instruments.

Should also fix StandardGST runs with instruments.
sandialabs · Sep 13, 2023 · 843d714 · 843d714
1 parent 3e98e17
commit 843d714
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/pygsti/models/explicitmodel.py b/pygsti/models/explicitmodel.py
@@ -1565,10 +1565,12 @@ def add_availability(opkey, op):
 
         if all([udim == 2 for udim in all_udims]):
             return _QubitProcessorSpec(nqudits, list(gate_unitaries.keys()), gate_unitaries, availability,
-                                       qubit_labels=qudit_labels)
+                                       qubit_labels=qudit_labels,
+                                       instrument_names=list(self.instruments.keys()), nonstd_instruments=self.instruments)
         else:
             return _QuditProcessorSpec(qudit_labels, all_udims, list(gate_unitaries.keys()), gate_unitaries,
-                                       availability)
+                                       availability,
+                                       instrument_names=list(self.instruments.keys()), nonstd_instruments=self.instruments)
 
     def create_modelmember_graph(self):
         return _MMGraph({

diff --git a/pygsti/models/modelconstruction.py b/pygsti/models/modelconstruction.py
@@ -936,13 +936,18 @@ def _spec_to_densevec(spec, is_prep):
                 inst_members = {}
                 for k, lst in instrument_spec.items():
                     member = None
-                    for effect_spec, prep_spec in lst:
-                        effect_vec = _spec_to_densevec(effect_spec, is_prep=False)
-                        prep_vec = _spec_to_densevec(prep_spec, is_prep=True)
-                        if member is None:
-                            member = _np.outer(effect_vec, prep_vec)
-                        else:
-                            member += _np.outer(effect_vec, prep_vec)
+                    if len(lst) == 2:
+                        for effect_spec, prep_spec in lst:
+                            effect_vec = _spec_to_densevec(effect_spec, is_prep=False)
+                            prep_vec = _spec_to_densevec(prep_spec, is_prep=True)
+                            if member is None:
+                                member = _np.outer(effect_vec, prep_vec)
+                            else:
+                                member += _np.outer(effect_vec, prep_vec)
+                    else: # elements are key, array of outer product already
+                        # TODO: This appears to be the new standard format. Deprecate outer prod code above?
+                        # But old code could still be useful.
+                        member = lst.copy()
 
                     assert (member is not None), \
                         "You must provide at least one rank-1 specifier for each instrument member!"