changing equality identity.

weinbe58 · weinbe58 · commit 02c35fd700be · 2023-09-27T21:07:19.000-04:00
diff --git a/src/bloqade/emulate/ir/space.py b/src/bloqade/emulate/ir/space.py
@@ -123,7 +123,7 @@ def transition_state_at(self, index: int, fro: int, to: int) -> NDArray:
         row_indices, col_config = self.atom_type.transition_state_at(
             self.configurations, index, fro, to
         )
-        if self.space_type == SpaceType.FullSpace:
+        if self.space_type is SpaceType.FullSpace:
             return (row_indices, col_config)
         else:
             col_indices = np.searchsorted(self.configurations, col_config)

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ def transition_state_at(self, index: int, fro: int, to: int) -> NDArray:`
`123`	`123`	`row_indices, col_config = self.atom_type.transition_state_at(`
`124`	`124`	`self.configurations, index, fro, to`
`125`	`125`	`)`
`126`		`- if self.space_type == SpaceType.FullSpace:`
	`126`	`+ if self.space_type is SpaceType.FullSpace:`
`127`	`127`	`return (row_indices, col_config)`
`128`	`128`	`else:`
`129`	`129`	`col_indices = np.searchsorted(self.configurations, col_config)`