minor bug fix for profiler

facebookresearch · Jan 31, 2022 · 54863fd · 54863fd
1 parent 3a8213a
commit 54863fd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/shaDow/profiler.py b/shaDow/profiler.py
@@ -101,9 +101,10 @@ def profile(self) -> None:
                 elif mg == 'sizes':
                     adj = sb.adj_ens[e]
                     args = [adj.indptr, subg_sizes, 'global']
-                self.value_metrics['global'][e][mg].append(
-                    getattr(self, f"_profile_{mg}")(*args)
-                )
+                if args is not None:
+                    self.value_metrics['global'][e][mg].append(
+                        getattr(self, f"_profile_{mg}")(*args)
+                    )
             # TODO handle local metrics
 
     def summarize(self):