diff --git a/ossdbs/electrodes/__init__.py b/ossdbs/electrodes/__init__.py index 5eb9cbe..f9217e7 100644 --- a/ossdbs/electrodes/__init__.py +++ b/ossdbs/electrodes/__init__.py @@ -75,10 +75,6 @@ from .microelectrode import MicroElectrodeModel, MicroElectrodeParameters from .neuro_pace import NeuroPaceModel, NeuroPaceParameters from .pins_medical import PINSMedicalModel, PINSMedicalParameters -from .pmt_seeg import ( - PMTsEEGModel, - PMTsEEGParameters, -) ELECTRODES = { "AbbottStJudeActiveTip6142_6145": AbbottStJudeActiveTip6142_6145, @@ -130,11 +126,11 @@ "DixiSEEG12Custom": DixiSEEGModel, "DixiSEEG15Custom": DixiSEEGModel, "DixiSEEG18Custom": DixiSEEGModel, - "PMTsEEG2102_08Custom": PMTsEEGModel, - "PMTsEEG2102_10Custom": PMTsEEGModel, - "PMTsEEG2102_12Custom": PMTsEEGModel, - "PMTsEEG2102_14Custom": PMTsEEGModel, - "PMTsEEG2102_16Custom": PMTsEEGModel, + "PMTsEEG2102_08Custom": DixiSEEGModel, + "PMTsEEG2102_10Custom": DixiSEEGModel, + "PMTsEEG2102_12Custom": DixiSEEGModel, + "PMTsEEG2102_14Custom": DixiSEEGModel, + "PMTsEEG2102_16Custom": DixiSEEGModel, "Medtronic3387Custom": MedtronicModel, "Medtronic3389Custom": MedtronicModel, "Medtronic3391Custom": MedtronicModel, @@ -159,7 +155,7 @@ "BostonScientificCartesiaXModel": BostonScientificCartesiaParameters, "BostonScientificCartesiaHXModel": BostonScientificCartesiaParameters, "DixiSEEGModel": DixiSEEGParameters, - "PMTsEEGModel": PMTsEEGParameters, + "PMTsEEGModel": DixiSEEGParameters, "MedtronicModel": MedtronicParameters, "MedtronicSenSightModel": MedtronicParameters, "MicroElectrodeModel": MicroElectrodeParameters, @@ -203,8 +199,6 @@ "PMTsEEG2102_12", "PMTsEEG2102_14", "PMTsEEG2102_16", - "PMTsEEGModel", - "PMTsEEGParameters", "Medtronic3387", "Medtronic3389", "Medtronic3391", diff --git a/ossdbs/electrodes/defaults.py b/ossdbs/electrodes/defaults.py index 36d7a69..54ceaf2 100644 --- a/ossdbs/electrodes/defaults.py +++ b/ossdbs/electrodes/defaults.py @@ -46,10 +46,6 @@ PINSMedicalModel, PINSMedicalParameters, ) -from .pmt_seeg import ( - PMTsEEGModel, - PMTsEEGParameters, -) default_electrode_parameters = { "AbbottStJudeActiveTip6146_6149": AbbottStJudeActiveTipParameters( @@ -252,7 +248,7 @@ total_length=400.0, n_contacts=18, ), - "PMTsEEG2102_08": PMTsEEGParameters( + "PMTsEEG2102_08": DixiSEEGParameters( tip_length=0.8, contact_length=2.0, contact_spacing=2.0, @@ -260,7 +256,7 @@ total_length=400.0, n_contacts=8, ), - "PMTsEEG2102_10": PMTsEEGParameters( + "PMTsEEG2102_10": DixiSEEGParameters( tip_length=0.8, contact_length=2.0, contact_spacing=2.0, @@ -268,7 +264,7 @@ total_length=400.0, n_contacts=10, ), - "PMTsEEG2102_12": PMTsEEGParameters( + "PMTsEEG2102_12": DixiSEEGParameters( tip_length=0.8, contact_length=2.0, contact_spacing=2.0, @@ -276,7 +272,7 @@ total_length=400.0, n_contacts=12, ), - "PMTsEEG2102_14": PMTsEEGParameters( + "PMTsEEG2102_14": DixiSEEGParameters( tip_length=0.8, contact_length=2.0, contact_spacing=2.0, @@ -284,7 +280,7 @@ total_length=400.0, n_contacts=14, ), - "PMTsEEG2102_16": PMTsEEGParameters( + "PMTsEEG2102_16": DixiSEEGParameters( tip_length=0.8, contact_length=2.0, contact_spacing=2.0, @@ -518,7 +514,7 @@ def PMTsEEG2102_08( ): """PMTsEEG2102_08 electrode.""" parameters = default_electrode_parameters["PMTsEEG2102_08"] - return PMTsEEGModel(parameters, rotation, direction, position) + return DixiSEEGModel(parameters, rotation, direction, position) def PMTsEEG2102_10( @@ -526,7 +522,7 @@ def PMTsEEG2102_10( ): """PMTsEEG2102_10 electrode.""" parameters = default_electrode_parameters["PMTsEEG2102_10"] - return PMTsEEGModel(parameters, rotation, direction, position) + return DixiSEEGModel(parameters, rotation, direction, position) def PMTsEEG2102_12( @@ -534,7 +530,7 @@ def PMTsEEG2102_12( ): """PMTsEEG2102_12 electrode.""" parameters = default_electrode_parameters["PMTsEEG2102_12"] - return PMTsEEGModel(parameters, rotation, direction, position) + return DixiSEEGModel(parameters, rotation, direction, position) def PMTsEEG2102_14( @@ -542,7 +538,7 @@ def PMTsEEG2102_14( ): """PMTsEEG2102_14 electrode.""" parameters = default_electrode_parameters["PMTsEEG2102_14"] - return PMTsEEGModel(parameters, rotation, direction, position) + return DixiSEEGModel(parameters, rotation, direction, position) def PMTsEEG2102_16( @@ -550,4 +546,4 @@ def PMTsEEG2102_16( ): """PMTsEEG2102_16 electrode.""" parameters = default_electrode_parameters["PMTsEEG2102_16"] - return PMTsEEGModel(parameters, rotation, direction, position) + return DixiSEEGModel(parameters, rotation, direction, position)