Remove GCN pos-rec

XENONnT · Nov 19, 2024 · c522cb5 · c522cb5
1 parent e222b25
commit c522cb5
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 72 deletions.
diff --git a/straxen/corrections_services.py b/straxen/corrections_services.py
@@ -17,17 +17,13 @@
 corrections_w_file = [
     "mlp_model",
     "cnn_model",
-    "gcn_model",
     "s2_xy_map_mlp",
     "s2_xy_map_cnn",
-    "s2_xy_map_gcn",
     "s2_xy_map",
     "s1_xyz_map_mlp",
     "s1_xyz_map_cnn",
-    "s1_xyz_map_gcn",
     "fdc_map_mlp",
     "fdc_map_cnn",
-    "fdc_map_gcn",
     "s1_aft_xyz_map",
     "bayes_model",
 ]
@@ -284,7 +280,7 @@ def get_config_from_cmt(self, run_id, model_type, version="ONLINE"):
         utilix.mongo_storage.MongoDownloader()
 
         :param run_id: run id from runDB
-        :param model_type: model type and neural network type; model_mlp, or model_gcn or model_cnn
+        :param model_type: model type and neural network type; model_mlp, or model_cnn
         :param version: version
         :param return: NN weights file name
 

diff --git a/straxen/plugins/events/__init__.py b/straxen/plugins/events/__init__.py
@@ -58,9 +58,6 @@
 from . import event_s2_positions_cnn
 from .event_s2_positions_cnn import *
 
-from . import event_s2_positions_gcn
-from .event_s2_positions_gcn import *
-
 from . import event_s1_positions_cnn
 from .event_s1_positions_cnn import *
 

diff --git a/straxen/plugins/events/event_s2_positions_gcn.py b/straxen/plugins/events/event_s2_positions_gcn.py
diff --git a/straxen/plugins/peaks/__init__.py b/straxen/plugins/peaks/__init__.py
@@ -22,9 +22,6 @@
 from . import peak_positions_cnf
 from .peak_positions_cnf import *
 
-from . import peak_positions_gcn
-from .peak_positions_gcn import *
-
 from . import peak_positions_mlp
 from .peak_positions_mlp import *
 

diff --git a/straxen/plugins/peaks/peak_positions_gcn.py b/straxen/plugins/peaks/peak_positions_gcn.py