From c4f1ad300b601b69e1203921c3b6c158cff8ae4c Mon Sep 17 00:00:00 2001
From: Pat Alt <55311242+pat-alt@users.noreply.github.com>
Date: Fri, 19 May 2023 10:57:22 +0200
Subject: [PATCH] removed flux from tested models because it requires too much
 test time

---
 src/conformal_models/conformal_models.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/conformal_models/conformal_models.jl b/src/conformal_models/conformal_models.jl
index 550f367..7ff73d0 100644
--- a/src/conformal_models/conformal_models.jl
+++ b/src/conformal_models/conformal_models.jl
@@ -113,14 +113,14 @@ const tested_atomic_models = Dict(
         :evo_tree => :(@load EvoTreeRegressor pkg = EvoTrees),
         :nearest_neighbor => :(@load KNNRegressor pkg = NearestNeighborModels),
         :light_gbm => :(@load LGBMRegressor pkg = LightGBM),
-        :neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
+        # :neural_network => :(@load NeuralNetworkRegressor pkg = MLJFlux),
     ),
     :classification => Dict(
         :logistic => :(@load LogisticClassifier pkg = MLJLinearModels),
         :evo_tree => :(@load EvoTreeClassifier pkg = EvoTrees),
         :nearest_neighbor => :(@load KNNClassifier pkg = NearestNeighborModels),
         :light_gbm => :(@load LGBMClassifier pkg = LightGBM),
-        :neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
+        # :neural_network => :(@load NeuralNetworkClassifier pkg = MLJFlux),
     ),
 )