Cleaned up return from volume.py::_from_numpy()

brainglobe · Nov 16, 2023 · 5de85ef · 5de85ef
1 parent 0dcdbee
commit 5de85ef
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/brainrender/actors/volume.py b/brainrender/actors/volume.py
@@ -78,13 +78,12 @@ def _from_numpy(self, griddata, voxel_size, color, **volume_kwargs):
         Creates a vedo.Volume actor from a 3D numpy array
         with volume data
         """
-        volume = VedoVolume(
+
+        return VedoVolume(
             griddata,
             spacing=[voxel_size, voxel_size, voxel_size],
             **volume_kwargs,
-        )
-        volume.cmap(color)
-        return volume
+        ).cmap(color)
 
     def _from_file(self, filepath, voxel_size, color, **volume_kwargs):
         """

diff --git a/examples/neurons.py b/examples/neurons.py
@@ -13,7 +13,7 @@
 scene = Scene(title="neurons")
 
 # Add a neuron from file
-# scene.add(Neuron("data/neuron1.swc"))
+scene.add(Neuron("data/neuron1.swc"))
 
 # Download neurons data with morphapi
 mlapi = MouseLightAPI()