add import for pkg_resources

sunlabuiuc · Feb 24, 2024 · 240f8e2 · 240f8e2
1 parent 6d2912f
commit 240f8e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhealth/sampler/sage_sampler.py b/pyhealth/sampler/sage_sampler.py
@@ -1,6 +1,6 @@
 import copy
 from typing import List, Optional, Tuple, NamedTuple, Union, Callable, TypeVar
-
+import pkg_resources
 import torch
 from torch import Tensor
 
@@ -207,4 +207,4 @@ def sample(self, batch):
         return {"image": image, "label": label, "batch_size": batch_size, "node_id": n_id, "adjacencies": adjs}
 
     def __repr__(self):
-        return '{}(sizes={})'.format(self.__class__.__name__, self.sizes)
+        return '{}(sizes={})'.format(self.__class__.__name__, self.sizes)