Fix formatting with pre-commit hooks

Author: neubig

.pre-commit-config.yaml

@@ -8,5 +8,4 @@ repos:
 -   repo: https://github.com/pre-commit/mirrors-mypy
     rev: v1.7.1
     hooks:
-    -   id: mypy
-        additional_dependencies: [types-all]
+    -   id: mypy

pyproject.toml

@@ -2,6 +2,9 @@
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
+[tool.hatch.build.targets.wheel]
+packages = ["src/starter_repo"]
+
 [project]
 name = "starter-repo"
 version = "0.1.0"

src/starter_repo/plot_data.py

@@ -1,36 +1,40 @@
 #!/usr/bin/env python3
 
-from pathlib import Path
 import argparse
-import pandas as pd
+from pathlib import Path
+from typing import List, Tuple
+
 import matplotlib.pyplot as plt
-from typing import list, tuple
+import pandas as pd
+
 
-def read_csv_data(file_path: Path, x_col: str, y_col: str) -> tuple[list[float], list[float]]:
+def read_csv_data(file_path: Path, x_col: str, y_col: str) -> Tuple[List[float], List[float]]:
     """Read data from a CSV file and return specified columns.
-    
+
     Args:
         file_path: Path to the CSV file
         x_col: Name of the column to use for x-axis
         y_col: Name of the column to use for y-axis
-        
+
     Returns:
         Tuple of x and y data as lists
     """
     df = pd.read_csv(file_path)
     return df[x_col].tolist(), df[y_col].tolist()
 
-def create_plot(x_data: list[float], y_data: list[float], 
-                x_label: str, y_label: str, title: str) -> plt.Figure:
+
+def create_plot(
+    x_data: List[float], y_data: List[float], x_label: str, y_label: str, title: str
+) -> plt.Figure:
     """Create a plot from the provided data.
-    
+
     Args:
         x_data: Data for x-axis
         y_data: Data for y-axis
         x_label: Label for x-axis
         y_label: Label for y-axis
         title: Plot title
-        
+
     Returns:
         matplotlib Figure object
     """
@@ -41,22 +45,30 @@ def create_plot(x_data: list[float], y_data: list[float],
     ax.set_title(title)
     return fig
 
+
 def main() -> None:
     parser = argparse.ArgumentParser(description="Create plots from CSV data")
     parser.add_argument("file_path", type=Path, help="Path to the CSV file")
     parser.add_argument("x_column", type=str, help="Column name for x-axis")
     parser.add_argument("y_column", type=str, help="Column name for y-axis")
-    parser.add_argument("--output", "-o", type=Path, default=Path("plot.png"),
-                       help="Output file path (default: plot.png)")
-    parser.add_argument("--title", "-t", type=str, default="Data Plot",
-                       help="Plot title (default: Data Plot)")
-    
+    parser.add_argument(
+        "--output",
+        "-o",
+        type=Path,
+        default=Path("plot.png"),
+        help="Output file path (default: plot.png)",
+    )
+    parser.add_argument(
+        "--title", "-t", type=str, default="Data Plot", help="Plot title (default: Data Plot)"
+    )
+
     args = parser.parse_args()
-    
+
     x_data, y_data = read_csv_data(args.file_path, args.x_column, args.y_column)
     fig = create_plot(x_data, y_data, args.x_column, args.y_column, args.title)
     fig.savefig(args.output)
     plt.close(fig)
 
+
 if __name__ == "__main__":
-    main()
+    main()

tests/test_plot_data.py

@@ -1,25 +1,26 @@
 from pathlib import Path
+
 import pandas as pd
 import pytest
-from starter_repo.plot_data import read_csv_data, create_plot
+from starter_repo.plot_data import create_plot, read_csv_data
+
 
 @pytest.fixture
 def sample_csv(tmp_path: Path) -> Path:
     """Create a sample CSV file for testing."""
-    df = pd.DataFrame({
-        'x': [1, 2, 3, 4, 5],
-        'y': [2, 4, 6, 8, 10]
-    })
+    df = pd.DataFrame({"x": [1, 2, 3, 4, 5], "y": [2, 4, 6, 8, 10]})
     file_path = tmp_path / "test.csv"
     df.to_csv(file_path, index=False)
     return file_path
 
+
 def test_read_csv_data(sample_csv: Path) -> None:
     """Test reading data from CSV file."""
-    x_data, y_data = read_csv_data(sample_csv, 'x', 'y')
+    x_data, y_data = read_csv_data(sample_csv, "x", "y")
     assert x_data == [1, 2, 3, 4, 5]
     assert y_data == [2, 4, 6, 8, 10]
 
+
 def test_create_plot() -> None:
     """Test plot creation."""
     x_data = [1, 2, 3]
@@ -30,4 +31,4 @@ def test_create_plot() -> None:
     ax = fig.axes[0]
     assert ax.get_xlabel() == "X"
     assert ax.get_ylabel() == "Y"
-    assert ax.get_title() == "Test Plot"
+    assert ax.get_title() == "Test Plot"