Skip to content

Commit

Permalink
Hide 'BlueConfig' type from allowed arguments
Browse files Browse the repository at this point in the history
  • Loading branch information
anilbey committed Feb 16, 2024
1 parent 133e0e9 commit e706c4e
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 7 deletions.
6 changes: 2 additions & 4 deletions bluecellulab/circuit/circuit_access/bluepy_circuit_access.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,17 +24,15 @@
import bluepy
from bluepy.enums import Cell as BluepyCell
from bluepy.impl.connectome_sonata import SonataConnectome
else:
BlueConfig = type("BlueConfig", (), {})

logger = logging.getLogger(__name__)


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):
Expand Down
5 changes: 2 additions & 3 deletions bluecellulab/circuit/config/bluepy_simulation_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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):
Expand Down

0 comments on commit e706c4e

Please sign in to comment.