fix: upgrade black to 26.1.0 and reformat all files

Author: Ritesh313

neon_tree_classification/inference/__init__.py

@@ -5,16 +5,16 @@
 
 Usage:
     from neon_tree_classification.inference import TreeClassifier
-    
+
     # Load from checkpoint
     classifier = TreeClassifier.from_checkpoint(
         checkpoint_path='path/to/model.ckpt',
         taxonomic_level='species'
     )
-    
+
     # Predict single image
     result = classifier.predict('path/to/image.jpg', top_k=5)
-    
+
     # Batch prediction
     results = classifier.predict_batch(['img1.jpg', 'img2.jpg'])
 """

neon_tree_classification/inference/model_registry.py

@@ -8,7 +8,6 @@
 from typing import Dict, Optional, List
 import warnings
 
-
 # Model catalog - will be populated with HuggingFace URLs later
 AVAILABLE_MODELS = {
     "resnet_species": {

neon_tree_classification/inference/preprocessing.py

@@ -12,7 +12,7 @@
 
 
 def load_image(
-    image_input: Union[str, Path, Image.Image, np.ndarray, torch.Tensor]
+    image_input: Union[str, Path, Image.Image, np.ndarray, torch.Tensor],
 ) -> Image.Image:
     """
     Load image from various input formats and convert to PIL Image.

pyproject.toml

@@ -160,3 +160,8 @@ exclude = [
 
 [tool.ruff.per-file-ignores]
 "__init__.py" = ["F401"]
+
+[dependency-groups]
+dev = [
+    "black>=25.0.0",
+]

scripts/get_dataloaders.py

@@ -36,7 +36,6 @@
 sys.path.append(str(Path(__file__).parent.parent))
 from neon_tree_classification.core.dataset import NeonCrownDataset
 
-
 DATASET_URL = "https://www.dropbox.com/scl/fi/v49xi6d7wtetctqphebx0/neon_tree_classification_dataset.zip?rlkey=fb7bz6kd0ckip4u0qd5xdor58&st=dvjyd5ry&dl=1"
 
 

scripts/upload_to_huggingface.py

@@ -43,7 +43,7 @@ def load_lightning_checkpoint(checkpoint_path: str) -> Dict[str, Any]:
 
 
 def extract_model_state_dict(
-    lightning_state_dict: Dict[str, torch.Tensor]
+    lightning_state_dict: Dict[str, torch.Tensor],
 ) -> Dict[str, torch.Tensor]:
     """Extract just the model weights from Lightning's state_dict.