fixing bug in gravity packing, adding F401 to ruff ignore list

PMEAL · Mar 12, 2024 · ebb2c3f · ebb2c3f
1 parent 502b761
commit ebb2c3f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -164,3 +164,5 @@ exclude = [".git", ".github", ".venv", "build", "docs", "examples", "test"]
 line-length = 92
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["E402", "F401", "F403"]
+[tool.ruff.lint]
+extend-ignore = ["F401"]
diff --git a/src/porespy/generators/_pseudo_packings.py b/src/porespy/generators/_pseudo_packings.py
@@ -210,7 +210,7 @@ def pseudo_gravity_packing(
     # Generate elevation values to initialize queue
     from porespy.generators import ramp
     h = ramp(im.shape, inlet=0, outlet=im.shape[0], axis=0)*mask
-    h = h*1.0 + np.random.rand(len(h))
+    h = h*1.0 + np.random.rand(*im.shape)
     tmp = np.arange(h.size)[mask.flatten()]
     inds = np.vstack(np.unravel_index(tmp, im.shape)).T
     h = h.flatten()[mask.flatten()]