From ceb8651c0b462a35ba0cd91b3cfad5f0b9c60dd3 Mon Sep 17 00:00:00 2001 From: selmanozleyen Date: Mon, 9 Sep 2024 15:06:10 +0200 Subject: [PATCH] remove redundant lines --- src/cfp/solvers/_genot.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/src/cfp/solvers/_genot.py b/src/cfp/solvers/_genot.py index 174f5d5..a287ad0 100644 --- 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)