Add ruff rules (#241)

Author: stefmolin

docs/conf.py

@@ -13,7 +13,7 @@
 import data_morph
 
 sys.path.insert(0, str(Path().absolute()))
-from post_build import determine_versions  # noqa: E402
+from post_build import determine_versions
 
 project = 'Data Morph'
 current_year = dt.date.today().year

pyproject.toml

@@ -102,14 +102,13 @@ lint.select = [
   "I",   # isort
   "N",   # pep8-naming
   "PTH", # flake8-use-pathlib
+  "RUF", # ruff-specific rules
   "SIM", # flake8-simplify
   "TRY", # tryceratops
   "UP",  # pyupgrade
   "W",   # pycodestyle warning
 ]
 lint.ignore = [
-  "ANN101", # missing type annotation for self (will be removed in future ruff version)
-  "ANN102", # missing type annotation for cls in classmethod (will be removed in future ruff version)
   "E501",   # line-too-long
   "TRY003", # avoid specifying long messages outside the exception class (revisit later and consider making custom exceptions)
 ]

src/data_morph/bounds/bounding_box.py

@@ -97,7 +97,7 @@ def __eq__(self, other: BoundingBox) -> bool:
     def __repr__(self) -> str:
         return '<BoundingBox>\n' f'  x={self.x_bounds}' '\n' f'  y={self.y_bounds}'
 
-    def adjust_bounds(self, x: Number = None, y: Number = None) -> None:
+    def adjust_bounds(self, x: Number | None = None, y: Number | None = None) -> None:
         """
         Adjust bounding box range.
 

src/data_morph/data/dataset.py

@@ -1,5 +1,7 @@
 """Class representing a dataset for morphing."""
 
+from __future__ import annotations
+
 from numbers import Number
 
 import matplotlib.pyplot as plt
@@ -39,13 +41,13 @@ class Dataset:
         Utility for creating :class:`Dataset` objects from CSV files.
     """
 
-    _REQUIRED_COLUMNS = ['x', 'y']
+    _REQUIRED_COLUMNS = ('x', 'y')
 
     def __init__(
         self,
         name: str,
         df: pd.DataFrame,
-        scale: Number = None,
+        scale: Number | None = None,
     ) -> None:
         self.df: pd.DataFrame = self._validate_data(df).pipe(self._scale_data, scale)
         """pandas.DataFrame: DataFrame containing columns x and y."""
@@ -181,7 +183,7 @@ def _validate_data(self, data: pd.DataFrame) -> pd.DataFrame:
 
     @plot_with_custom_style
     def plot(
-        self, ax: Axes = None, show_bounds: bool = True, title: str = 'default'
+        self, ax: Axes | None = None, show_bounds: bool = True, title: str = 'default'
     ) -> Axes:
         """
         Plot the dataset and its bounds.

src/data_morph/data/loader.py

@@ -1,9 +1,12 @@
 """Load data for morphing."""
 
+from __future__ import annotations
+
 from importlib.resources import files
 from itertools import zip_longest
 from numbers import Number
 from pathlib import Path
+from typing import ClassVar
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -41,7 +44,7 @@ class DataLoader:
     """
 
     _DATA_PATH: str = 'data/starter_shapes/'
-    _DATASETS: dict = {
+    _DATASETS: ClassVar[dict[str, str]] = {
         'bunny': 'bunny.csv',
         'cat': 'cat.csv',
         'dino': 'dino.csv',
@@ -66,7 +69,7 @@ def __init__(self) -> None:
     def load_dataset(
         cls,
         dataset: str,
-        scale: Number = None,
+        scale: Number | None = None,
     ) -> Dataset:
         """
         Load dataset.

src/data_morph/shapes/bases/line_collection.py

@@ -1,5 +1,7 @@
 """Base class for shapes that are composed of lines."""
 
+from __future__ import annotations
+
 from collections.abc import Iterable
 from numbers import Number
 
@@ -97,7 +99,7 @@ def distance(self, x: Number, y: Number) -> float:
         )
 
     @plot_with_custom_style
-    def plot(self, ax: Axes = None) -> Axes:
+    def plot(self, ax: Axes | None = None) -> Axes:
         """
         Plot the shape.
 

src/data_morph/shapes/bases/point_collection.py

@@ -1,5 +1,7 @@
 """Base class for shapes that are composed of points."""
 
+from __future__ import annotations
+
 from collections.abc import Iterable
 from numbers import Number
 
@@ -52,7 +54,7 @@ def distance(self, x: Number, y: Number) -> float:
         )
 
     @plot_with_custom_style
-    def plot(self, ax: Axes = None) -> Axes:
+    def plot(self, ax: Axes | None = None) -> Axes:
         """
         Plot the shape.
 

src/data_morph/shapes/bases/shape.py

@@ -103,7 +103,7 @@ def _recursive_repr(self, attr: str | None = None) -> str:
         )
 
     @abstractmethod
-    def plot(self, ax: Axes = None) -> Axes:
+    def plot(self, ax: Axes | None = None) -> Axes:
         """
         Plot the shape.
 

src/data_morph/shapes/circles.py

@@ -1,5 +1,7 @@
 """Shapes that are circular in nature."""
 
+from __future__ import annotations
+
 from numbers import Number
 
 import matplotlib.pyplot as plt
@@ -33,7 +35,7 @@ class Circle(Shape):
         The radius of the circle.
     """
 
-    def __init__(self, dataset: Dataset, radius: Number = None) -> None:
+    def __init__(self, dataset: Dataset, radius: Number | None = None) -> None:
         self.center: np.ndarray = dataset.df[['x', 'y']].mean().to_numpy()
         """numpy.ndarray: The (x, y) coordinates of the circle's center."""
 
@@ -63,7 +65,7 @@ def distance(self, x: Number, y: Number) -> float:
         )
 
     @plot_with_custom_style
-    def plot(self, ax: Axes = None) -> Axes:
+    def plot(self, ax: Axes | None = None) -> Axes:
         """
         Plot the shape.
 
@@ -159,7 +161,7 @@ def distance(self, x: Number, y: Number) -> float:
         )
 
     @plot_with_custom_style
-    def plot(self, ax: Axes = None) -> Axes:
+    def plot(self, ax: Axes | None = None) -> Axes:
         """
         Plot the shape.
 

src/data_morph/shapes/factory.py

@@ -2,6 +2,7 @@
 
 from itertools import zip_longest
 from numbers import Number
+from typing import ClassVar
 
 import matplotlib.pyplot as plt
 import numpy as np
@@ -56,7 +57,7 @@ class ShapeFactory:
         The starting dataset to morph into other shapes.
     """
 
-    _SHAPE_MAPPING: dict = {
+    _SHAPE_MAPPING: ClassVar[dict[str, type[Shape]]] = {
         'bullseye': Bullseye,
         'circle': Circle,
         'high_lines': HighLines,