chore: generalize function

Opentrons · May 1, 2024 · 1f996ae · 1f996ae
1 parent 91ec95f
commit 1f996ae
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test-data-generation/src/test_data_generation/deck_configuration/strategies.py b/test-data-generation/src/test_data_generation/deck_configuration/strategies.py
@@ -122,10 +122,10 @@ def a_column(
     )
 
 
-def _above_or_below_is_module_or_trash(col: Column, hs_slot: Slot) -> bool:
-    """Return True if the deck has a module above or below the heater shaker."""
-    above = col.slot_above(hs_slot)
-    below = col.slot_below(hs_slot)
+def _above_or_below_is_module_or_trash(col: Column, slot: Slot) -> bool:
+    """Return True if the deck has a module above or below the specified slot."""
+    above = col.slot_above(slot)
+    below = col.slot_below(slot)
 
     return (above is not None and above.contents.is_module_or_trash_bin()) or (
         below is not None and below.contents.is_module_or_trash_bin()