Merge pull request #273 from sunlabuiuc/20240224-neighborsampler-bug

add import for pkg_resources

Author: ycq091044

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)