Format Python code with Black

- Auto-formatted all Python files to conform to Black standards
- Fixes CI formatting check failures
- No functional changes, only formatting

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: dshkol

pycancensus/__init__.py

@@ -13,29 +13,36 @@
 from .regions import list_census_regions, search_census_regions
 from .vectors import list_census_vectors, search_census_vectors
 from .datasets import list_census_datasets
-from .settings import set_api_key, get_api_key, set_cache_path, get_cache_path, remove_api_key, show_api_key
+from .settings import (
+    set_api_key,
+    get_api_key,
+    set_cache_path,
+    get_cache_path,
+    remove_api_key,
+    show_api_key,
+)
 from .geometry import get_census_geometry
 from .cache import list_cache, remove_from_cache, clear_cache
 from .hierarchy import parent_census_vectors, child_census_vectors, find_census_vectors
 
 __all__ = [
     "get_census",
-    "list_census_regions", 
+    "list_census_regions",
     "search_census_regions",
     "list_census_vectors",
-    "search_census_vectors", 
+    "search_census_vectors",
     "list_census_datasets",
     "set_api_key",
     "get_api_key",
     "remove_api_key",
     "show_api_key",
-    "set_cache_path", 
+    "set_cache_path",
     "get_cache_path",
     "get_census_geometry",
     "list_cache",
     "remove_from_cache",
     "clear_cache",
     "parent_census_vectors",
-    "child_census_vectors", 
+    "child_census_vectors",
     "find_census_vectors",
-]
+]

pycancensus/cache.py

@@ -16,35 +16,35 @@
 def get_cached_data(cache_key: str) -> Optional[Any]:
     """
     Retrieve data from cache if it exists.
-    
+
     Parameters
     ----------
     cache_key : str
         Unique identifier for the cached data.
-        
+
     Returns
     -------
     Any or None
         Cached data if found, None otherwise.
     """
     cache_path = Path(get_cache_path())
     cache_file = cache_path / f"{cache_key}.pkl"
-    
+
     if cache_file.exists():
         try:
             with open(cache_file, "rb") as f:
                 return pickle.load(f)
         except Exception:
             # If cache file is corrupted, remove it
             cache_file.unlink(missing_ok=True)
-            
+
     return None
 
 
 def cache_data(cache_key: str, data: Any) -> None:
     """
     Cache data to disk.
-    
+
     Parameters
     ----------
     cache_key : str
@@ -54,9 +54,9 @@ def cache_data(cache_key: str, data: Any) -> None:
     """
     cache_path = Path(get_cache_path())
     cache_path.mkdir(parents=True, exist_ok=True)
-    
+
     cache_file = cache_path / f"{cache_key}.pkl"
-    
+
     try:
         with open(cache_file, "wb") as f:
             pickle.dump(data, f)
@@ -67,7 +67,7 @@ def cache_data(cache_key: str, data: Any) -> None:
 def list_cache() -> pd.DataFrame:
     """
     List all cached data files.
-    
+
     Returns
     -------
     pd.DataFrame
@@ -77,66 +77,71 @@ def list_cache() -> pd.DataFrame:
         - size_mb: File size in MB
         - created: Creation timestamp
         - modified: Last modification timestamp
-        
+
     Examples
     --------
     >>> import pycancensus as pc
     >>> cache_list = pc.list_cache()
     >>> print(cache_list)
     """
     cache_path = Path(get_cache_path())
-    
+
     if not cache_path.exists():
-        return pd.DataFrame(columns=["cache_key", "file_path", "size_mb", "created", "modified"])
-    
+        return pd.DataFrame(
+            columns=["cache_key", "file_path", "size_mb", "created", "modified"]
+        )
+
     cache_files = []
-    
+
     for cache_file in cache_path.glob("*.pkl"):
         try:
             stat = cache_file.stat()
-            cache_files.append({
-                "cache_key": cache_file.stem,
-                "file_path": str(cache_file),
-                "size_mb": round(stat.st_size / (1024 * 1024), 2),
-                "created": pd.Timestamp.fromtimestamp(stat.st_ctime),
-                "modified": pd.Timestamp.fromtimestamp(stat.st_mtime),
-            })
+            cache_files.append(
+                {
+                    "cache_key": cache_file.stem,
+                    "file_path": str(cache_file),
+                    "size_mb": round(stat.st_size / (1024 * 1024), 2),
+                    "created": pd.Timestamp.fromtimestamp(stat.st_ctime),
+                    "modified": pd.Timestamp.fromtimestamp(stat.st_mtime),
+                }
+            )
         except Exception:
             continue
-    
+
     return pd.DataFrame(cache_files)
 
 
-def remove_from_cache(cache_keys: Optional[List[str]] = None, 
-                      all_cache: bool = False) -> None:
+def remove_from_cache(
+    cache_keys: Optional[List[str]] = None, all_cache: bool = False
+) -> None:
     """
     Remove items from cache.
-    
+
     Parameters
     ----------
     cache_keys : list of str, optional
         Specific cache keys to remove. If None and all_cache=False,
         does nothing.
     all_cache : bool, default False
         If True, removes all cached data.
-        
+
     Examples
     --------
     >>> import pycancensus as pc
     >>> # Remove specific cache entries
     >>> pc.remove_from_cache(["regions_CA16", "vectors_CA16"])
-    >>> 
+    >>>
     >>> # Remove all cache (use with caution!)
     >>> pc.remove_from_cache(all_cache=True)
     """
     cache_path = Path(get_cache_path())
-    
+
     if not cache_path.exists():
         print("No cache directory found.")
         return
-    
+
     removed_count = 0
-    
+
     if all_cache:
         # Remove all .pkl files
         for cache_file in cache_path.glob("*.pkl"):
@@ -145,9 +150,9 @@ def remove_from_cache(cache_keys: Optional[List[str]] = None,
                 removed_count += 1
             except Exception as e:
                 print(f"Warning: Failed to remove {cache_file}: {e}")
-        
+
         print(f"Removed {removed_count} cached files.")
-        
+
     elif cache_keys:
         # Remove specific cache keys
         for cache_key in cache_keys:
@@ -161,7 +166,7 @@ def remove_from_cache(cache_keys: Optional[List[str]] = None,
                     print(f"Warning: Failed to remove {cache_key}: {e}")
             else:
                 print(f"Cache key not found: {cache_key}")
-        
+
         if removed_count > 0:
             print(f"Removed {removed_count} cached files.")
     else:
@@ -171,12 +176,12 @@ def remove_from_cache(cache_keys: Optional[List[str]] = None,
 def clear_cache() -> None:
     """
     Clear all cached data.
-    
+
     This is an alias for remove_from_cache(all_cache=True).
-    
+
     Examples
     --------
     >>> import pycancensus as pc
     >>> pc.clear_cache()
     """
-    remove_from_cache(all_cache=True)
+    remove_from_cache(all_cache=True)