diff --git a/scripts/gene_expression.py b/scripts/gene_expression.py index 6cdf15c..ea46765 100644 --- a/scripts/gene_expression.py +++ b/scripts/gene_expression.py @@ -9,5 +9,8 @@ gexp = watlas.get_gene_expression(gene_names=["npr-1","egl-4"],neuron_ids=["ADAR","ADEL"],th=4) print(gexp) +gexp = watlas.get_gene_expression(gene_names=["egl-4"],neuron_ids="ADAL",th=4) +print(gexp) + gexp = watlas.cengen.get_expression(gene_wbids=["WBGene00001173"],neuron_ids=["ADA","ADE","ADF","ADL","AFD"],th=2) print(gexp) diff --git a/wormneuroatlas/NeuroAtlas.py b/wormneuroatlas/NeuroAtlas.py index 431831b..5c0c086 100644 --- a/wormneuroatlas/NeuroAtlas.py +++ b/wormneuroatlas/NeuroAtlas.py @@ -1953,6 +1953,9 @@ def get_gene_expression(self, neuron_ais=None, neuron_ids=None, if neuron_ais is not None: neuron_cis = self.cengen_is[neuron_ais] elif neuron_ids is not None: + if type(neuron_ids) == str: + # See https://github.com/francescorandi/wormneuroatlas/issues/4 + neuron_ids = [neuron_ids] try: neuron_ids[0] except: neuron_ids = np.array([neuron_ids]) neuron_ais = self.ids_to_ais(neuron_ids)