Skip to content

Commit

Permalink
fixing tests
Browse files Browse the repository at this point in the history
  • Loading branch information
miili committed Nov 5, 2023
1 parent 38c4163 commit fba8556
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions lassie/tracers/fast_marching/velocity_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,21 +51,27 @@ class VelocityModel3D(BaseModel):

def model_post_init(self, __context: Any) -> None:
grid_spacing = self.grid_spacing
neast = int(round(self.east_size / grid_spacing)) + 1
nnorth = int(round(self.north_size / grid_spacing)) + 1
ndepth = int(round(self.depth_size / grid_spacing)) + 1

self._east_coords = np.arange(
self._east_coords = np.linspace(
self.east_bounds[0],
self.east_bounds[1],
grid_spacing,
neast,
endpoint=True,
)
self._north_coords = np.arange(
self._north_coords = np.linspace(
self.north_bounds[0],
self.north_bounds[1],
grid_spacing,
nnorth,
endpoint=True,
)
self._depth_coords = np.arange(
self._depth_coords = np.linspace(
self.depth_bounds[0],
self.depth_bounds[1],
grid_spacing,
ndepth,
endpoint=True,
)

self._velocity_model = np.zeros(
Expand Down

0 comments on commit fba8556

Please sign in to comment.