fix: annotate umap weighted edges for mypy

Author: lmeyerov

CHANGELOG.md

@@ -8,6 +8,9 @@ This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm
 ## [Development]
 <!-- Do Not Erase This Section - Used for tracking unreleased changes -->
 
+### Fixed
+- **Mypy**: Declare UMAP mixin weighted-edge DataFrame attributes for type checking.
+
 ## [0.50.5 - 2026-01-25]
 
 ### Fixed

graphistry/umap_utils.py

@@ -250,6 +250,9 @@ class UMAPMixin(MIXIN_BASE):
     """
     UMAP Mixin for automagic UMAPing
     """
+    _weighted_edges_df: Optional[DataFrameLike]
+    _weighted_edges_df_from_nodes: Optional[DataFrameLike]
+    _weighted_edges_df_from_edges: Optional[DataFrameLike]
 
     def __init__(self, *a, **kw):
         super().__init__(*a, **kw)