diff --git a/python/cugraph/cugraph/gnn/feature_storage/feat_storage.py b/python/cugraph/cugraph/gnn/feature_storage/feat_storage.py
index 6d24a768f11..77a53882fc4 100644
--- a/python/cugraph/cugraph/gnn/feature_storage/feat_storage.py
+++ b/python/cugraph/cugraph/gnn/feature_storage/feat_storage.py
@@ -164,7 +164,9 @@ def get_data(
             )
 
         feat = self.fd[feat_name][type_name]
-        if not isinstance(wgth, MissingModule) and isinstance(feat, wgth.WholeMemoryEmbedding):
+        if not isinstance(wgth, MissingModule) and isinstance(
+            feat, wgth.WholeMemoryEmbedding
+        ):
             indices_tensor = (
                 indices
                 if isinstance(indices, torch.Tensor)