From fe1a3eee000240a225d2399ee490ccdddbfffe2f Mon Sep 17 00:00:00 2001
From: Mason Protter <mason.protter@icloud.com>
Date: Thu, 1 Aug 2024 14:39:10 +0200
Subject: [PATCH] remove diag transformation from `generate_diffusion_function`

---
 src/systems/diffeqs/sdesystem.jl | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/src/systems/diffeqs/sdesystem.jl b/src/systems/diffeqs/sdesystem.jl
index cac295d0f2..439e525670 100644
--- a/src/systems/diffeqs/sdesystem.jl
+++ b/src/systems/diffeqs/sdesystem.jl
@@ -277,9 +277,6 @@ function generate_diffusion_function(sys::SDESystem, dvs = unknowns(sys),
     if isdde
         eqs = delay_to_function(sys, eqs)
     end
-    if eqs isa AbstractMatrix && __num_isdiag_noise(eqs)
-        eqs = __get_num_diag_noise(eqs)
-    end
     u = map(x -> time_varying_as_func(value(x), sys), dvs)
     p = if has_index_cache(sys) && get_index_cache(sys) !== nothing
         reorder_parameters(get_index_cache(sys), ps)