Remove unneeded option

jacobbieker · Jul 8, 2020 · b0b0cdc · b0b0cdc
1 parent 4c0fe07
commit b0b0cdc
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/factnn/generator/pytorch/eventfile_dataloader.py b/factnn/generator/pytorch/eventfile_dataloader.py
@@ -96,7 +96,7 @@ def get(self, idx):
 
 class EventFileDiffuseDataset(Dataset):
 
-    def __init__(self, root, split="trainval", transform=None, pre_transform=None, num_points=0):
+    def __init__(self, root, split="trainval", transform=None, pre_transform=None):
         """
         EventFile Dataloader for specifically Disp calculations,
         only using the diffuse gamma sources that have the extra information
@@ -106,7 +106,6 @@ def __init__(self, root, split="trainval", transform=None, pre_transform=None, n
         :param num_points: The number of points to have, either using points multiple times, or subselecting from the total points
         """
         self.processed_filenames = []
-        self.num_points = num_points
         self.split = split
         super(EventFileDiffuseDataset, self).__init__(root, transform, pre_transform)