Small formatting fixes

README.md

@@ -71,10 +71,11 @@ The position of the center of the plane is given by a set of `(x, y, z)` coordin
 Whe using one of the named orientation, you don't need to pass a whole set of `(x, y, z)` coordinates for the plane center. A single value is sufficient as the other two won't affect the plane position:
 
 ```python
-f = bgh.heatmap(
+f = bgh.Heatmap(
     values,
     position=1000,
-    orientation="sagittal",  # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
+    orientation="sagittal",
+    # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
     thickness=1000,
     atlas_name="allen_cord_20um",
     format='2D',
@@ -83,11 +84,13 @@ f = bgh.heatmap(
 ```
 
 Also, you can create a slice with a plane centered in the brain by passing `position=None`:
+
 ```python
-f = bgh.heatmap(
+f = bgh.Heatmap(
     values,
     position=None,
-    orientation="sagittal",  # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
+    orientation="sagittal",
+    # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
     thickness=1000,
     atlas_name="mpin_zfish_1um",
     format='2D',
@@ -100,14 +103,15 @@ Once happy with the position of the slicing planes, creating a visualization is
 
 ```python
 
-bgh.heatmap(
+bgh.Heatmap(
     values,
     position=(
         8000,
         5000,
         5000,
     ),
-    orientation="horizontal",  # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
+    orientation="horizontal",
+    # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)
     title="horizontal view",
     vmin=-5,
     vmax=3,

brainglobe_heatmap/__init__.py

@@ -6,6 +6,6 @@
     # package is not installed
     pass
 
-from brainglobe_heatmap.heatmaps import heatmap
+from brainglobe_heatmap.heatmaps import Heatmap
 from brainglobe_heatmap.planner import plan
 from brainglobe_heatmap.slicer import get_structures_slice_coords

brainglobe_heatmap/heatmaps.py

@@ -48,7 +48,7 @@
     return vmax, vmin
 
 
-class heatmap:
+class Heatmap:
     def __init__(
         self,
         values: Dict,

brainglobe_heatmap/planner.py

@@ -16,7 +16,7 @@
 from rich.table import Table
 from vedo import Arrow, Sphere
 
-from brainglobe_heatmap.heatmaps import heatmap
+from brainglobe_heatmap.heatmaps import Heatmap
 from brainglobe_heatmap.plane import Plane
 
 settings.BACKGROUND_COLOR = amber_lighter
@@ -47,7 +47,7 @@
     )
 
 
-class plan(heatmap):
+class plan(Heatmap):
     def __init__(
         self,
         regions: Union[dict, list],
@@ -116,38 +116,3 @@
 
         self.scene.render(interactive=self.interactive, zoom=self.zoom)
         return self.scene
-
-
-if __name__ == "__main__":
-    regions = dict(  # scalar values for each region
-        TH=1,
-        RSP=0.2,
-        AI=0.4,
-        SS=-3,
-        MO=2.6,
-        PVZ=-4,
-        LZ=-3,
-        VIS=2,
-        AUD=0.3,
-        RHP=-0.2,
-        STR=0.5,
-        CB=0.5,
-        FRP=-1.7,
-        HIP=3,
-        PA=-4,
-    )
-
-    plan(
-        regions,
-        # position of the center point of the plane
-        position=(
-            8000,
-            5000,
-            5000,
-        ),
-        # or 'sagittal', or 'horizontal' or a tuple (x,y,z)
-        orientation="frontal",
-        # thickness of the slices used for rendering (in microns)
-        thickness=2000,
-        arrow_scale=750,
-    ).show()

brainglobe_heatmap/slicer.py

@@ -136,7 +136,7 @@
         Slices the meshes in a 3D brainrender scene using the gien planes
         """
         # slice the scene
-        for n, plane in enumerate((self.plane0, self.plane1)):
+        for _, plane in enumerate((self.plane0, self.plane1)):
             scene.slice(plane, actors=regions, close_actors=True)
 
         scene.slice(self.plane0, actors=scene.root, close_actors=False)

examples/cellfinder_cell_density.py

@@ -4,14 +4,18 @@
     density (count/volume) of cells in each brain region
 """
 
+from pathlib import Path
+
 import pandas as pd
 from brainglobe_atlasapi.bg_atlas import BrainGlobeAtlas
 from brainrender._io import load_mesh_from_file
 
 import brainglobe_heatmap as bgh
 
 # get the number of cells for each region
-data = pd.read_hdf("examples/cell_counts_example.h5")
+cells_path = Path(__file__).parent / "cell_counts_example.h5"
+
+data = pd.read_hdf(cells_path)
 cell_counts = data.groupby("region").count()
 
 # get regions two levels up the hierarchy
@@ -39,7 +43,7 @@
 print(cell_counts)
 
 
-f = bgh.heatmap(
+f = bgh.Heatmap(
     cell_counts.density.to_dict(),
     position=(
         8000,

examples/heatmap_2d.py

@@ -23,7 +23,7 @@
 )
 
 
-f = bgh.heatmap(
+f = bgh.Heatmap(
     values,
     # when using a named orientation, you can pass a single value!
     position=5000,

examples/heatmap_3d.py

@@ -23,7 +23,7 @@
 )
 
 
-scene = bgh.heatmap(
+scene = bgh.Heatmap(
     values,
     position=(8000, 5000, 5000),
     orientation="frontal",  # or 'sagittal', or 'horizontal' or a tuple (x,y,z)

examples/heatmap_human_brain.py

@@ -10,7 +10,7 @@
 values = dict(SFG=1, PrCG=2, Ca=4, Pu=10)  # scalar values for each region
 
 
-scene = bgh.heatmap(
+scene = bgh.Heatmap(
     values,
     position=(100000, 100000, 100000),
     thickness=1000,

examples/heatmap_spinal_cord.py

@@ -19,7 +19,7 @@
 }
 
 
-f = bgh.heatmap(
+f = bgh.Heatmap(
     values,
     position=1000,
     # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)

examples/heatmap_zebrafish.py

@@ -20,7 +20,7 @@
     "cerebellum": 0.5,
 }
 
-f = bgh.heatmap(
+f = bgh.Heatmap(
     values,
     position=175,
     # 'frontal' or 'sagittal', or 'horizontal' or a tuple (x,y,z)

pyproject.toml

@@ -78,7 +78,7 @@ exclude = ["__init__.py", "build", ".eggs"]
 fix = true
 
 [tool.ruff.lint]
-select = ["I", "E", "F"]
+select = ["I", "E", "F", "B"]
 
 [tool.tox]
 legacy_tox_ini = """