Merge pull request #799 from dotunv/feat/optimization-issues-765-766-767-768

feat: parallelize snapshots, add graph cache, optimise backfill memory, enrich artifact metadata

Author: gelluisaac

astroml/cache/__init__.py

@@ -1,7 +1,8 @@
-"""Redis caching layer for AstroML.
+"""Caching layer for AstroML.
 
-This module provides Redis-based caching for frequently accessed data including:
+This module provides caching for frequently accessed data including:
 - Graph snapshots
+- Graph computation results (adjacency lists, edge/node features)
 - Feature computation results
 - Model predictions
 - Artifact metadata
@@ -11,11 +12,19 @@
 - Cache invalidation on data updates
 - Cache hit/miss metrics
 - Decorator-based caching
+- Per-data-version and window graph computation caching (issue #767)
 """
 
 from __future__ import annotations
 
 from astroml.cache.decorators import cache_feature_store
+from astroml.cache.graph_cache import (
+    GraphComputationCache,
+    GraphCacheConfig,
+    GraphCacheStats,
+    get_graph_cache,
+    invalidate_graph_cache,
+)
 from astroml.cache.redis_cache import (
     CacheConfig,
     CacheStats,
@@ -41,4 +50,10 @@
     "invalidate_cache",
     "get_cache_stats",
     "clear_all_caches",
+    # Graph computation cache (issue #767)
+    "GraphComputationCache",
+    "GraphCacheConfig",
+    "GraphCacheStats",
+    "get_graph_cache",
+    "invalidate_graph_cache",
 ]