Merge pull request #106 from theislab/refactor/remove_redundant_lines

remove redundant lines
theislab · Sep 9, 2024 · d6b0af2 · d6b0af2
2 parents e4da834 + ceb8651
commit d6b0af2
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/src/cfp/solvers/_genot.py b/src/cfp/solvers/_genot.py
@@ -186,11 +186,7 @@ def step_fn(
         rng = jax.random.split(rng, 5)
         rng, rng_resample, rng_noise, rng_time, rng_step_fn = rng
 
-        src, tgt = batch["src_cell_data"], batch["tgt_cell_data"]
         condition = batch.get("condition")
-
-        matching_data = (src, tgt)
-
         (src, tgt), matching_data = self._prepare_data(batch)
         n = src.shape[0]
         time = self.time_sampler(rng_time, n)