style: ruff formatting

noamteyssier · Oct 9, 2024 · 660148a · 660148a
1 parent 7ea12c4
commit 660148a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/muxsim/gen.py b/src/muxsim/gen.py
@@ -35,7 +35,8 @@ def _gen_moi(self) -> np.ndarray:
     def _gen_assignments(self) -> list:
         """Generate assignments for each cell/guide pair."""
         assignment = [
-            np.random.choice(self.num_guides, self.moi[i]) for i in range(self.num_cells)
+            np.random.choice(self.num_guides, self.moi[i])
+            for i in range(self.num_cells)
         ]
         return assignment
 

diff --git a/tests/test_muxsim.py b/tests/test_muxsim.py
@@ -14,7 +14,8 @@ def test_init():
     assert ms.num_guides == NUM_GUIDES
     assert ms.umi_sums.size == NUM_CELLS
     assert ms.moi.size == NUM_CELLS
-    assert len(ms.assignments)== NUM_CELLS
+    assert len(ms.assignments) == NUM_CELLS
+
 
 def test_gen():
     ms = MuxSim(num_cells=1000, num_guides=100)