diff --git a/bluecellulab/circuit/circuit_access/bluepy_circuit_access.py b/bluecellulab/circuit/circuit_access/bluepy_circuit_access.py index c91d2917..5fe99172 100644 --- a/bluecellulab/circuit/circuit_access/bluepy_circuit_access.py +++ b/bluecellulab/circuit/circuit_access/bluepy_circuit_access.py @@ -24,8 +24,6 @@ import bluepy from bluepy.enums import Cell as BluepyCell from bluepy.impl.connectome_sonata import SonataConnectome -else: - BlueConfig = type("BlueConfig", (), {}) logger = logging.getLogger(__name__) @@ -33,8 +31,8 @@ class BluepyCircuitAccess: """Bluepy implementation of CircuitAccess protocol.""" - def __init__(self, simulation_config: str | Path | BlueConfig | BluepySimulationConfig) -> None: - """Initialize bluepy circuit object.""" + def __init__(self, simulation_config: str | Path | BluepySimulationConfig) -> None: + """Initialize bluepy circuit object. BlueConfig also is a valid type.""" if not BLUEPY_AVAILABLE: raise ExtraDependencyMissingError("bluepy") if isinstance(simulation_config, Path): diff --git a/bluecellulab/circuit/config/bluepy_simulation_config.py b/bluecellulab/circuit/config/bluepy_simulation_config.py index de16880a..a5b0bf2f 100644 --- a/bluecellulab/circuit/config/bluepy_simulation_config.py +++ b/bluecellulab/circuit/config/bluepy_simulation_config.py @@ -9,8 +9,6 @@ if BLUEPY_AVAILABLE: from bluepy_configfile.configfile import BlueConfig from bluepy.utils import open_utf8 -else: - BlueConfig = type("BlueConfig", (), {}) from bluecellulab.circuit.config.sections import Conditions, ConnectionOverrides from bluecellulab.stimuli import Stimulus @@ -20,7 +18,8 @@ class BluepySimulationConfig: """Bluepy implementation of SimulationConfig protocol.""" _connection_overrides: list[ConnectionOverrides] = [] - def __init__(self, config: str | BlueConfig) -> None: + def __init__(self, config: str) -> None: + """a str or a BlueConfig object are valid.""" if not BLUEPY_AVAILABLE: raise ExtraDependencyMissingError("bluepy") if isinstance(config, str):