update inverse indexing for multidimensional inversion grids - needs testing
@@ -75,18 +75,18 @@ class Inversion:
# We're using the integration grid to preseed, not fix values. So
# Track the grid, but don't fix the integration grid points
self.inds_fixed = []
- self.inds_preseed = [self.fm.statevec.index(k) for k in
- self.integration_grid.keys()]
- self.inds_free = [i for i in np.arange(self.fm.nstate, dtype=int) if
- not (i in self.inds_fixed)]
+ self.inds_preseed = np.array([self.fm.statevec.index(k) for k in
+ self.integration_grid.keys()])
+ self.inds_free = np.array([i for i in np.arange(self.fm.nstate, dtype=int) if
+ not (i in self.inds_fixed)])
else:
# We're using the integration grid to fix values. So
# Get set up to fix the integration grid points
- self.inds_fixed = [self.fm.statevec.index(k) for k in
- self.integration_grid.keys()]
- self.inds_free = [i for i in np.arange(self.fm.nstate, dtype=int) if
- not (i in self.inds_fixed)]
+ self.inds_fixed = np.array([self.fm.statevec.index(k) for k in
+ self.integration_grid.keys()])
+ self.inds_free = np.array([i for i in np.arange(self.fm.nstate, dtype=int) if
+ not (i in self.inds_fixed)])
self.inds_preseed = []
self.x_fixed = None
